backward(mone)的问题

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()就可以了

你可能感兴趣的:(深度学习,pytorch)