mmengine.model.stack_batch

mmengine.model.stack_batch(tensor_list, pad_size_divisor=1, pad_value=0)

Stack multiple tensors to form a batch and pad the tensor to the max shape use the right bottom padding mode in these images. If pad_size_divisor > 0, add padding to ensure the shape of each dim is divisible by pad_size_divisor.

Parameters

• tensor_list (List[Tensor]) – A list of tensors with the same dim.

• pad_size_divisor (int) – If pad_size_divisor > 0, add padding to ensure the shape of each dim is divisible by pad_size_divisor. This depends on the model, and many models need to be divisible by 32. Defaults to 1

• pad_value (int, float) – The padding value. Defaults to 0.

Returns

The n dim tensor.

Return type

Tensor