


Hooks
YES
Mostly YES. Only outputs of submodules
NO
NO












Table 1: The pros (or cons) of some of the existing methods for feature extraction with PyTorch
In the next section of this article, let’s see how we can get YES across the board.
FX to The Rescue