tensorplay.nn.init
Functions
calculate_gain() [source]
python
calculate_gain(nonlinearity, param=None)constant_() [source]
python
constant_(tensor, val)dirac_() [source]
python
dirac_(tensor, groups=1)eye_() [source]
python
eye_(tensor)kaiming_normal_() [source]
python
kaiming_normal_(tensor, a=0, mode='fan_in', nonlinearity='leaky_relu')kaiming_uniform_() [source]
python
kaiming_uniform_(tensor, a=0, mode='fan_in', nonlinearity='leaky_relu')normal_() [source]
python
normal_(tensor, mean=0.0, std=1.0)ones_() [source]
python
ones_(tensor)uniform_() [source]
python
uniform_(tensor, a=0.0, b=1.0)xavier_normal_() [source]
python
xavier_normal_(tensor, gain=1.0)xavier_uniform_() [source]
python
xavier_uniform_(tensor, gain=1.0)zeros_() [source]
python
zeros_(tensor)