View source on GitHub
|
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. |