Grayscale — Torchvision main documentation - pytorch.org
pytorch.org › vision › mainGrayscale (num_output_channels = 1) [source] ¶ Convert image to grayscale. If the image is torch Tensor, it is expected to have […, 3, H, W] shape, where … means an arbitrary number of leading dimensions. Parameters. num_output_channels – (1 or 3) number of channels desired for output image. Returns. Grayscale version of the input.