Shortcuts

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

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.