Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

tfm.vision.preprocess_ops.clip_or_pad_to_fixed_size

Pads data to a fixed length at the first dimension.

tfm.vision.preprocess_ops.clip_or_pad_to_fixed_size(
    input_tensor, size, constant_values=0
)

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.