Pytorch 数学计算（只能将一个元素张量转换为 Python 标量）

Question

在这种情况下，只有一个元素张量可以转换为 Python 标量是什么意思，我该如何调试它？

x1 = (max-min)*torch.rand(1, 21) + min
x2 = (max-min)*torch.rand(1, 21) + min
zipped_list = zip(x1, x2)
y = [math.sin(2*x1+2) * math.cos(0.5*x2)+0.5 for (x1, x2) in zipped_list]

输出

ValueError: only one element tensors can be converted to Python scalars

Answer 1

您会收到该错误，因为您的火炬张量（x1 和 x2）不是单元素张量。

t = torch.tensor([10, 20])
print(t.item()) # This will throw an error since the tensor has more than 1 element

t = torch.tensor([10])
t.item() # This will print 10

要解决您的问题，请将math.sin和math.cos调用替换为torch.sin和torch.cos 。 torch.sin或torch.cos计算所有元素的正弦或余弦值。

Answer 2

问题是因为使用math.cos和math.sin 。 math.cos 不是矢量化的，但 np.cos 是。 使用np.sin和np.cos或torch.sin和torch.cos 。