Unverified Commit 0a946d5b authored by Ambuj Pawar's avatar Ambuj Pawar Committed by GitHub
Browse files

DOC: add limitation of decode_jpeg in the function docstring (#6637)



* add limitation of decode_jpeg in the function docstring

* undo unwanted changes in decode_png
Co-authored-by: default avatarAmbuj Pawar <your_email@abc.example>
parent 4c66712f
...@@ -137,7 +137,9 @@ def decode_jpeg( ...@@ -137,7 +137,9 @@ def decode_jpeg(
the raw bytes of the JPEG image. This tensor must be on CPU, the raw bytes of the JPEG image. This tensor must be on CPU,
regardless of the ``device`` parameter. regardless of the ``device`` parameter.
mode (ImageReadMode): the read mode used for optionally mode (ImageReadMode): the read mode used for optionally
converting the image. Default: ``ImageReadMode.UNCHANGED``. converting the image. The supported modes are: ``ImageReadMode.UNCHANGED``,
``ImageReadMode.GRAY`` and ``ImageReadMode.RGB``
Default: ``ImageReadMode.UNCHANGED``.
See ``ImageReadMode`` class for more information on various See ``ImageReadMode`` class for more information on various
available modes. available modes.
device (str or torch.device): The device on which the decoded image will device (str or torch.device): The device on which the decoded image will
......
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