"vscode:/vscode.git/clone" did not exist on "e2db55eda3f806b21eca595a790b80a2a60ab9ab"
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 ...@@ -209,6 +209,7 @@ Conversion
v2.ToTensor v2.ToTensor
PILToTensor PILToTensor
v2.PILToTensor v2.PILToTensor
v2.ToImageTensor
ConvertImageDtype ConvertImageDtype
v2.ConvertImageDtype v2.ConvertImageDtype
v2.ConvertDtype v2.ConvertDtype
......
...@@ -27,6 +27,13 @@ class PILToTensor(Transform): ...@@ -27,6 +27,13 @@ class PILToTensor(Transform):
class ToImageTensor(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) _transformed_types = (is_simple_tensor, PIL.Image.Image, np.ndarray)
def _transform( 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