Asserts that weights satisfy constraints.
tfl.kronecker_factored_lattice_lib.assert_constraints(
weights, units, scale, monotonicities, eps=1e-06
)
Args |
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.
|
eps
|
Allowed constraints violation.
|
Returns |
List of assertion ops in graph mode or directly executes assertions in eager
mode.
|