tfl.kronecker_factored_lattice_lib.bias_initializer

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.

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

Kronecker-Factored Lattice bias of shape: (units).