Clarify TVTensor shapes

[mantasu]

📚 The doc issue

TVTensor is documented as being a subclass of torch.Tensor specifically for transforms. Perhaps it could be added that they only constrain the shape of torch.Tensor?

For example, I only use the utility functions of transforms.v2 and, when to_image referred me to Image, I still had to check the source code to confirm it is just a torch.Tensor of shape (C x H x W).

Suggest a potential alternative/fix

No response

[NicolasHug]

Hi @mantasu , we've added the main relevant info about TVTensors in this tutorial https://pytorch.org/vision/main/auto_examples/transforms/plot_transforms_getting_started.html#what-are-tvtensors which is linked to as the main entrypoint for all things related to TVTensors / TransformsV2. Do you think it already contains the info you were looking for, or should we edit it a bit?

[mantasu]

My only suggestion would be to add somewhere that TVTensors are just extensions of torch.Tensor with specified shape constraints. But it is just a preference and the tutorial already helps a lot, so this could be safely closed~