View source on GitHub
|
Returns function which does lattice interpolation.
tfl.test_utils.get_linear_lattice_interpolation_fn(
lattice_sizes,
monotonicities,
output_min,
output_max
)
Returned function matches lattice_layer.LinearInitializer with corresponding parameters.
Args:
lattice_sizes: list or tuple of integers which represents lattice sizes.monotonicities: monotonicity constraints.output_min: minimum output of linear function.output_max: maximum output of linear function.
Returns:
Function which takes d-dimension point and performs lattice interpolation assuming lattice weights are such that lattice represents linear function with given output_min and output_max. All monotonic dimesions of this linear function cotribute with same weight despite of numer of vertices per dimension. All non monotonic dimensions have weight 0.0.