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