• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1.. role:: hidden
2    :class: hidden-section
3
4.. _nn-init-doc:
5
6torch.nn.init
7=============
8
9.. warning::
10    All the functions in this module are intended to be used to initialize neural network
11    parameters, so they all run in :func:`torch.no_grad` mode and will not be taken into
12    account by autograd.
13
14.. currentmodule:: torch.nn.init
15.. autofunction:: calculate_gain
16.. autofunction:: uniform_
17.. autofunction:: normal_
18.. autofunction:: constant_
19.. autofunction:: ones_
20.. autofunction:: zeros_
21.. autofunction:: eye_
22.. autofunction:: dirac_
23.. autofunction:: xavier_uniform_
24.. autofunction:: xavier_normal_
25.. autofunction:: kaiming_uniform_
26.. autofunction:: kaiming_normal_
27.. autofunction:: trunc_normal_
28.. autofunction:: orthogonal_
29.. autofunction:: sparse_
30