Converts images/masks to PyTorch Tensors, inheriting from BasicTransform.
For images:
Converts HWC format to PyTorch CHW format
transpose_maskIf True, transposes 3D input mask dimensions from [height, width, num_channels] to
[num_channels, height, width].
pProbability of applying the transform. Default: 1.0.
>>> transform = ToTensorV2(transpose_mask=True)