get_image_num_channels
-
torchvision.transforms.functional.get_image_num_channels(img: torch.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
