ToTensor¶
- class torchvision.transforms.v2.ToTensor[source]¶
[BETA] [DEPRECATED] Use
v2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
instead.Convert a PIL Image or ndarray to tensor and scale the values accordingly.
Note
The ToTensor transform is in Beta stage, and while we do not expect disruptive breaking changes, some APIs may slightly change according to user feedback. Please submit any feedback you may have in this issue: https://github.com/pytorch/vision/issues/6753.
Warning
v2.ToTensor
is deprecated and will be removed in a future release. Please use insteadv2.Compose([v2.ToImage(), v2.ToDtype(torch.float32, scale=True)])
.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, CMYK, 1) or if the numpy.ndarray has dtype = np.uint8
In the other cases, tensors are returned without scaling.
Note
Because the input image is scaled to [0.0, 1.0], this transformation should not be used when transforming target image masks. See the references for implementing the transforms for image masks.