get_dimensions

torchvision.transforms.functional.get_dimensions(img: Tensor) → List[int]

Returns the dimensions of an image as [channels, height, width].

Parameters:

img (PIL Image or Tensor) – The image to be checked.

Returns:

The image dimensions.

Return type:

List[int]