Docs for torch.utils.hooks missing in Pytorch

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…


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.