Shortcuts

get_image_num_channels

torchvision.transforms.functional.get_image_num_channels(img: Tensor) int[source]

Returns the number of channels of an image.

Parameters:

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

Returns:

The number of channels.

Return type:

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