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

update nn.init doc to reflect the no_grad (#80882)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/80839

Pull Request resolved: https://github.com/pytorch/pytorch/pull/80882
Approved by: https://github.com/jbschlosser
This commit is contained in:
albanD
2022-07-07 17:19:29 +00:00
committed by PyTorch MergeBot
parent bd75b2fea1
commit 6f1d99b79f
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docs/source/nn.init.rst
View File

@ -6,6 +6,11 @@
torch.nn.init
=============
.. warning::
All the functions in this module are intended to be used to initialize neural network
parameters, so they all run in :func:`torch.no_grad` mode and will not be taken into
account by autograd.
.. currentmodule:: torch.nn.init
.. autofunction:: calculate_gain
.. autofunction:: uniform_