Unverified Commit be798eff authored by vfdev's avatar vfdev Committed by GitHub
Browse files

Updated _type_conversion.py docs (#7324)


Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent 449cc090
......@@ -209,6 +209,7 @@ Conversion
v2.ToTensor
PILToTensor
v2.PILToTensor
v2.ToImageTensor
ConvertImageDtype
v2.ConvertImageDtype
v2.ConvertDtype
......
......@@ -27,6 +27,13 @@ class PILToTensor(Transform):
class ToImageTensor(Transform):
"""[BETA] Convert a tensor or an ndarray or PIL Image to :class:`~torchvision.datapoints.Image`.
.. betastatus:: ToImageTensor transform
This transform does not support torchscript.
"""
_transformed_types = (is_simple_tensor, PIL.Image.Image, np.ndarray)
def _transform(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment