Initializes bias depending on output_min and output_max.
tfl.kronecker_factored_lattice_lib.bias_initializer(
units, output_min, output_max, dtype=tf.float32
)
If both output_min and output_max are set, bias is initialized to their
average. If only output_min is set, bias is initialized to output_min. If only
output_max is set, bias is initialized to output_max. Otherwise bias is
initialized to zeros.
Args |
units
|
Output dimension of the layer. Each of units bias will be initialized
identically.
|
output_min
|
None or minimum layer output.
|
output_max
|
None or maximum layer output.
|
dtype
|
dtype
|
Returns |
Kronecker-Factored Lattice bias of shape: (units) .
|