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