Stay organized with collections Save and categorize content based on your preferences.

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).