Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
Why do you think that https://github.com/ashishpatel26/Tools-to-Design-or-Visualize-Architecture-of-Neural-Network is a good alternative to torchlens