View source on GitHub
|
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
)
Args | |
|---|---|
input_tensor
|
Tensor with any dimension.
|
size
|
int number for the first dimension of output Tensor.
|
constant_values
|
int value assigned to the paddings.
|
Returns | |
|---|---|
Tensor with the first dimension padded to size.
|