tfl.kronecker_factored_lattice_lib.assert_constraints

Asserts that weights satisfy constraints.

tfl.kronecker_factored_lattice_lib.assert_constraints(
    weights, units, scale, monotonicities, output_min, output_max, eps=1e-06
)

weights KroneckerFactoredLattice weights tensor of shape: (1, lattice_sizes, units * dims, num_terms).
units Number of units per input dimension.
scale Scale variable of shape: (units, num_terms).
monotonicities Monotonicity constraints.
output_min None or minimum layer output.
output_max None or maximum layer output.
eps Allowed constraints violation.

List of assertion ops in graph mode or directly executes assertions in eager mode.