![]() |
Constraints for tfl.layers.KroneckerFactoredLattice
layer.
tfl.kronecker_factored_lattice_layer.KroneckerFactoredLatticeConstraints(
units, scale, monotonicities=None, output_min=None, output_max=None
)
Applies all constraints to the Kronecker-Factored Lattice weights. See
tfl.layers.KroneckerFactoredLattice
for more details.
Args | |
---|---|
units
|
Same meaning as corresponding parameter of
KroneckerFactoredLattice .
|
scale
|
Scale variable of shape: (units, num_terms) .
|
monotonicities
|
Same meaning as corresponding parameter of
KroneckerFactoredLattice .
|
output_min
|
Same meaning as corresponding parameter of
KroneckerFactoredLattice .
|
output_max
|
Same meaning as corresponding parameter of
KroneckerFactoredLattice .
|
Methods
get_config
get_config()
Standard Keras config for serialization.
__call__
__call__(
w
)
Applies constraints to w
.
Args | |
---|---|
w
|
Kronecker-Factored Lattice weights tensor of shape: (1, lattice_sizes,
units * dims, num_terms) .
|
Returns | |
---|---|
Constrained and projected w. |