convert_image_dtype¶
-
torchvision.transforms.functional.
convert_image_dtype
(image: torch.Tensor, dtype: torch.dtype = torch.float32) → torch.Tensor[source]¶ Convert a tensor image to the given
dtype
and scale the values accordingly This function does not support PIL Image.- Parameters
image (torch.Tensor) – Image to be converted
dtype (torch.dpython:type) – Desired data type of the output
- Returns
Converted image
- Return type
Tensor
Note
When converting from a smaller to a larger integer
dtype
the maximum values are not mapped exactly. If converted back and forth, this mismatch has no effect.- Raises
RuntimeError – When trying to cast
torch.float32
totorch.int32
ortorch.int64
as well as for trying to casttorch.float64
totorch.int64
. These conversions might lead to overflow errors since the floating pointdtype
cannot store consecutive integers over the whole range of the integerdtype
.
Examples using
convert_image_dtype
: