mmengine.model.stack_batch¶

mmengine.model.stack_batch(tensor_list, pad_size_divisor=1, pad_value=0)[source]¶

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