View source on GitHub

Returns a uniformly random sampled monotonic lattice layer weight tensor.

  • The uniform random monotonic function will initilaize the lattice parameters uniformly at random and make it such that the parameters are monotonically increasing for each input.
  • The random parameters will be sampled from [output_min, output_max]

lattice_sizes List or tuple of integers which represents lattice sizes.
output_min Minimum output of lattice layer after initialization.
output_max Maximum output of lattice layer after initialization.
units Output dimension of the layer. Each of units lattices will be initialized identically.
dtype dtype.

Lattice weights tensor of shape: (prod(lattice_sizes), units).