torch_mist.utils.shape

Module Contents

Functions

expand_to_same_shape(→ Tuple[torch.Tensor, Ellipsis])

torch_mist.utils.shape.expand_to_same_shape(*tensors) → Tuple[torch.Tensor, Ellipsis]