tfm.vision.preprocess_ops.clip_or_pad_to_fixed_size

Pads data to a fixed length at the first dimension.

input_tensor Tensor with any dimension.
size int number for the first dimension of output Tensor.
constant_values int value assigned to the paddings.

Tensor with the first dimension padded to size.