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代码示例展示了如何创建张量并进行基本的数学运算,包括取负数和计算平均值。示例中还涉及了张量的数据类型和设备属性,有助于初学者理解张量的基本概念及其在GPU上的应用。
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()就可以了
2051
5188

被折叠的 条评论
为什么被折叠?