Normalize异常报错

在使用Tensor`规范化处理的时候报下面这个错

from torchvision import transforms
import numpy as np
import torchvision
import torch

data = np.random.randint(0, 255, size=12)
img = data.reshape(2,2,3)


print(img)
print("*"*100)
transform1 = transforms.Compose([
    transforms.ToTensor(), # range [0, 255] -> [0.0,1.0]
    transforms.Normalize(mean = (10,10,10), std = (1,1,1)),
    ]
)
# img = img.astype('float')
norm_img = transform1(img) #进行规范化处理
print(norm_img)

添加上这句就可以，其实是给设置元素类型，看上面的提示也是这个意思