frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[easy] Expose documentation for a few global nn.Module hooks (#97185)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97185
Approved by: https://github.com/albanD
This commit is contained in:
Mikayla Gawarecki
2023-03-20 21:09:51 +00:00
committed by PyTorch MergeBot
parent 7a93865c46
commit b04363ead4
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/nn.rst
View File

@ -51,7 +51,11 @@ Global Hooks For Module
register_module_forward_pre_hook
register_module_forward_hook
register_module_backward_hook
register_module_full_backward_pre_hook
register_module_full_backward_hook
register_module_buffer_registration_hook
register_module_module_registration_hook
register_module_parameter_registration_hook
.. currentmodule:: torch