![]() |
Represents a kronecker-factored lattice layer.
tfl.model_info.KroneckerFactoredLatticeNode(
input_nodes, weights, scale, bias
)
Attributes | |
---|---|
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) .
|