nn.init.rst 518 Bytes
Newer Older
yuguo's avatar
yuguo committed
1
oneflow.nn.init
yuguo's avatar
yuguo committed
2
3
4
5
===============

.. The documentation is referenced from: 
   https://pytorch.org/docs/1.10/nn.init.html
yuguo's avatar
yuguo committed
6

yuguo's avatar
yuguo committed
7
8
9
10
11
12
13
14
15
16
17
18
19
.. currentmodule:: oneflow.nn.init
.. autofunction:: calculate_gain
.. autofunction:: uniform_
.. autofunction:: normal_
.. autofunction:: constant_
.. autofunction:: ones_
.. autofunction:: zeros_
.. autofunction:: xavier_uniform_
.. autofunction:: xavier_normal_
.. autofunction:: kaiming_uniform_
.. autofunction:: kaiming_normal_
.. autofunction:: trunc_normal_
.. autofunction:: orthogonal_