Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand


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