mmengine.model.stack_batch — mmengine 0.10.7 documentation (original) (raw)

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. Ifpad_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 bypad_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