Table of Contents

Shortcuts

get_dimensions¶

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

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]

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources