tfl.kronecker_factored_lattice_lib.scale_initializer

Initializes scale to alternate between 1 and -1 for each term.

units Output dimension of the layer. Each of units scale will be initialized identically.
num_terms Number of independently trained submodels per unit, the outputs of which are averaged to get the final output.

Kronecker-Factored Lattice scale of shape: (units, num_terms).