Table of Contents
Shortcuts

# convert_image_dtype

torchvision.transforms.functional.convert_image_dtype(image: Tensor, dtype: dtype = torch.float32) → Tensor

  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` to `torch.int32` or `torch.int64` as well as for trying to cast `torch.float64` to `torch.int64`. These conversions might lead to overflow errors since the floating point `dtype` cannot store consecutive integers over the whole range of the integer `dtype`.

## Docs

Access comprehensive developer documentation for PyTorch

View Docs

## Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

## Resources

Find development resources and get your questions answered

View Resources

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.