Docs for torch.utils.hooks missing in Pytorch

C++
1

I am new to PyTorch so apologies if my question is very basic
I am trying to implement some visualization techniques from different papers and I observed that hooks really play a very important role in this.

But when I wanted to find the documentation for torch.utils.hooks, I could not find it anywhere.
Is it because it’s not a completely developed feature of PyTorch or they ignored it since it is not a very important topic to be in the docs?

Thanks in advance…

2

Hi,

torch.utils.hooks are just internal utility functions that we use to implement hooks but that’s it.
You can find Tensor hooks by looking for register_hook() on Tensor.
You can find nn.Module hook by looking for register_forward_hook(), register_pre_forward_hook() on nn.Module.