MatchingData.set_matching_dimension

MatchingData.set_matching_dimension(target_batched=False, template_batched=False)

Configure batch dimensions for target and template

Parameters:

target_batchedbool, optional

Whether the target has a leading batch dimension.

template_batchedbool, optional

Whether the template has a leading batch dimension.

Notes

The batch dimension, if present, is always at position 0. When either side is batched, the other gets a singleton leading dimension so that both output shapes share the same number of dimensions.

Examples

>>> matching_data.set_matching_dimension(target_batched=True)