30 lines
778 B
ReStructuredText
30 lines
778 B
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
.. _nn-init-doc:
|
|
|
|
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_
|
|
.. autofunction:: normal_
|
|
.. autofunction:: constant_
|
|
.. autofunction:: ones_
|
|
.. autofunction:: zeros_
|
|
.. autofunction:: eye_
|
|
.. autofunction:: dirac_
|
|
.. autofunction:: xavier_uniform_
|
|
.. autofunction:: xavier_normal_
|
|
.. autofunction:: kaiming_uniform_
|
|
.. autofunction:: kaiming_normal_
|
|
.. autofunction:: trunc_normal_
|
|
.. autofunction:: orthogonal_
|
|
.. autofunction:: sparse_
|