/torchlens

Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Watchers