torchvision.transforms.ToTensor() can't work

1

import sys
print(sys.version)
import torchvision
print(torchvision.version)
import numpy as np

a = np.array([[[255,255,255]]])
b = torchvision.transforms.ToTensor()(a)
b

output:
3.6.5 |Anaconda, Inc.| (default, Mar 29 2018, 13:32:41) [MSC v.1900 64 bit (AMD64)]
0.3.0

tensor([[[255]],

    [[255]],

    [[255]]], dtype=torch.int32)
2

Try to pass an np.uint8 array, if you would like to get a normalized output tensor.

3

It works, thank you.