nerva_tensorflow.weight_initializers
Weight and bias initialization helpers for linear layers.
Functions
|
Normal (Gaussian) initialization with given mean and std. |
|
Uniform initialization within [a, b). |
Set all bias values to zero. |
|
|
Initialize a layer's parameters according to a named scheme. |
He / Kaiming normal initialization (for ReLU). |
|
He / Kaiming uniform initialization (for ReLU). |
|
|
Normal (Gaussian) initialization with given mean and std. |
|
Uniform initialization within [a, b). |
Xavier / Glorot normal initialization. |
|
Xavier / Glorot uniform initialization. |
|
Initialize weights to zero. |
- nerva_tensorflow.weight_initializers.set_bias_zero(b: tensorflow.Tensor)[source]
Set all bias values to zero.
- nerva_tensorflow.weight_initializers.set_bias_uniform(b: tensorflow.Tensor, a: float = 0.0, b_: float = 1.0)[source]
Uniform initialization within [a, b).
- nerva_tensorflow.weight_initializers.set_bias_normal(b: tensorflow.Tensor, mean: float = 0.0, std: float = 1.0)[source]
Normal (Gaussian) initialization with given mean and std.
- nerva_tensorflow.weight_initializers.set_weights_uniform(W: tensorflow.Tensor, a: float = 0.0, b: float = 1.0)[source]
Uniform initialization within [a, b).
- nerva_tensorflow.weight_initializers.set_weights_normal(W: tensorflow.Tensor, mean: float = 0.0, std: float = 1.0)[source]
Normal (Gaussian) initialization with given mean and std.
- nerva_tensorflow.weight_initializers.set_weights_xavier_uniform(W: tensorflow.Tensor)[source]
Xavier / Glorot uniform initialization.
- nerva_tensorflow.weight_initializers.set_weights_xavier_normal(W: tensorflow.Tensor)[source]
Xavier / Glorot normal initialization.
- nerva_tensorflow.weight_initializers.set_weights_he_normal(W: tensorflow.Tensor)[source]
He / Kaiming normal initialization (for ReLU).
- nerva_tensorflow.weight_initializers.set_weights_he_uniform(W: tensorflow.Tensor)[source]
He / Kaiming uniform initialization (for ReLU).