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
)

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.

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