Computes weights for PWL calibration.
tfl.pwl_calibration_lib.compute_interpolation_weights(
inputs, keypoints, lengths
)
Args |
inputs
|
Tensor of shape: (D0, D1, ..., DN, 1) which represents inputs to
to the pwl function. A typical shape is: (batch_size, 1) .
|
keypoints
|
Rank-1 tensor of shape (num_keypoints - 1) which represents
left keypoint of pieces of piecewise linear function along X axis.
|
lengths
|
Rank-1 tensor of shape (num_keypoints - 1) which represents
lengths of pieces of piecewise linear function along X axis.
|
Returns |
Interpolation weights tensor of shape: (D0, D1, ..., DN, num_keypoints) .
|