tfl.lattice_lib.assert_constraints

Asserts that weights satisfy constraints.

tfl.lattice_lib.assert_constraints(
    weights,
    lattice_sizes,
    monotonicities,
    edgeworth_trusts,
    trapezoid_trusts,
    monotonic_dominances,
    range_dominances,
    joint_monotonicities,
    joint_unimodalities,
    output_min=None,
    output_max=None,
    eps=1e-06
)

Args
weights	Lattice weights tensor of shape: (prod(lattice_sizes), units).
lattice_sizes	List or tuple of integers which represents lattice sizes.
monotonicities	Monotonicity constraints.
edgeworth_trusts	Edgeworth trust constraints.
trapezoid_trusts	Trapezoid trust constraints.
monotonic_dominances	Monotonic dominance constraints.
range_dominances	Range dominance constraints.
joint_monotonicities	Joint monotonicity constraints.
joint_unimodalities	Joint unimodality constraints.
output_min	None or lower bound constraints.
output_max	None or upper bound constraints.
eps	Allowed constraints violation.

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