I’ve seen that the hook methods are only available for classes that inherit from nn.Module. Is there any easy/simple way to use/do something like the hooks but in functions such as torch.cat or torch.nn.functional.interpolate? I’m particularly interested in tracing the output of these functions.