[fbsync] Add error message when setting 1D tensor ToImage() (#9114)

Summary: Co-authored-by: Nicolas Hug <[email protected]>

Reviewed By: AntoineSimoulin

Differential Revision: D79175021

fbshipit-source-id: a4345beae9801f31b30f6394452f07adf7bcf4fb

Author: NicolasHug

torchvision/tv_tensors/_image.py

@@ -43,7 +43,7 @@ def __new__(
 
         tensor = cls._to_tensor(data, dtype=dtype, device=device, requires_grad=requires_grad)
         if tensor.ndim < 2:
-            raise ValueError
+            raise ValueError(f"Tensor must be 2D or higher, got {tensor.ndim}D tensor.")
         elif tensor.ndim == 2:
             tensor = tensor.unsqueeze(0)