ConvertImageDtype¶
-
class
torchvision.transforms.
ConvertImageDtype
(dtype: torch.dtype)[source]¶ Convert a tensor image to the given
dtype
and scale the values accordingly This function does not support PIL Image.- Parameters
dtype (torch.dpython:type) – Desired data type of the output
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
ConvertImageDtype
:-
forward
(image)[source]¶ Defines the computation performed at every call.
Should be overridden by all subclasses.
Note
Although the recipe for forward pass needs to be defined within this function, one should call the
Module
instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.