fix docs in ToTensor

This commit is contained in:
liaoxingyu 2020-06-10 19:06:03 +08:00
parent 896933bf5e
commit 96b9ad2d99

View File

@ -20,7 +20,7 @@ class ToTensor(object):
"""Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor.
Converts a PIL Image or numpy.ndarray (H x W x C) in the range
[0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0]
[0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 255.0]
if the PIL Image belongs to one of the modes (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1)
or if the numpy.ndarray has dtype = np.uint8