Save the date! Google I/O returns May 18-20 Register now

tfl.model_info.KroneckerFactoredLatticeNode

Represents a kronecker-factored lattice layer.

tfl.model_info.KroneckerFactoredLatticeNode(
    input_nodes, weights, scale, bias
)

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