Module init

Module init 

Source

Enums§

FanMode
Fan computation mode for Kaiming initialization.

Functions§

constant
Initialize a tensor with a constant value.
kaiming_normal
Kaiming (He) normal initialization.
kaiming_uniform
Kaiming (He) uniform initialization.
normal
Initialize a tensor from a normal distribution N(mean, std).
ones
Initialize a tensor with all ones (as a variable).
uniform
Initialize a tensor from a uniform distribution U(low, high).
xavier_normal
Xavier (Glorot) normal initialization.
xavier_uniform
Xavier (Glorot) uniform initialization.
zeros
Initialize a tensor with all zeros (as a variable).