ToTensor — Torchvision main documentation - pytorch.org
pytorch.org/vision/master/generated/torchvision.transforms.ToTensor.htmlToTensor¶ class torchvision.transforms. ToTensor [source] ¶. Convert a PIL Image or numpy.ndarray to tensor. This transform does not support torchscript. Converts a PIL Image or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0] if the PIL Image belongs to one of the modes (L, LA, P, I, F, RGB, YCbCr, RGBA, …