one = torch.FloatTensor([1])
mone = one * -1
print(mone)
print(mone.mean())
tensor([-1.], device='cuda:0')
tensor(-1., device='cuda:0')
可以看到结果,两者数据类型不一致,所以mone改为mone.mean()就可以了
one = torch.FloatTensor([1])
mone = one * -1
print(mone)
print(mone.mean())
tensor([-1.], device='cuda:0')
tensor(-1., device='cuda:0')
可以看到结果,两者数据类型不一致,所以mone改为mone.mean()就可以了