View source on GitHub |
Bilinear resizes the images.
tfm.vision.spatial_transform_ops.bilinear_resize(
images: tf.Tensor, output_size: tf.Tensor
) -> tf.Tensor
Args | |
---|---|
images
|
A tensor in shape (batch_size, input_h, input_w, ...) with arbitrary numbers of channel dimensions. |
output_size
|
The size of the output image in (output_h, output_w). |
Returns | |
---|---|
A tensor in shape (batch_size, output_h, output_w, ...). The result has the same dtype as the input if it's float32, float16, bfloat16, otherwise the result is float32. |