johnmarktaylor91/torchlens
Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
PythonGPL-3.0
Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
PythonGPL-3.0