View source on GitHub
|
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).
|