ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Represents a kronecker-factored lattice layer.

input_nodes List of input nodes to the kronecker-factored lattice layer.
weights Kronecker-factored lattice kernel parameters of shape (1, lattice_sizes, units * dims, num_terms).
scale Kronecker-factored lattice scale parameters of shape (units, num_terms).
bias Kronecker-factored lattice bias parameters of shape (units).