Back to Oneflow

oneflow.nn.init

docs/source/nn.init.rst

1.0.0517 B
Original Source

oneflow.nn.init

.. The documentation is referenced from: https://pytorch.org/docs/1.10/nn.init.html

.. 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_