问题代码:
def exp_sum(self, x1, x2, x3, x4):
sum = math.exp(x1) + math.exp(x2) + math.exp(x3) + math.exp(x4)
return x1 / sum, x2 / sum, x3 / sum, x4 / sum
x1等都是tensors。
出现问题:
ValueError: only one element tensors can be converted to Python scalars
解决过程:
使用a =torch.randn(1, 1, 1)
生成一个只有一个元素的三维tensor,然后使用math.exp(a)
报错。
但是显然x1等的tensors不可能都是一个元素,因此我在网上查询这个错误,发现有将tensor转换成numpy的,然后将x1等转换成numpy,在使用math.exp()
,出现新的错误:
TypeError: only size-1 arrays can be converted to Python scalars
经过自己的试错发现,如果是numpy类型的数据,可以使用np.exp(),如果是tensor数据,可以使用torch.exp()。代码如下:
import torch
import numpy as np
a = torch.randn(3, 3, 3)
b = a.numpy()
c = torch.exp(a)
d = np.exp(b)
出来的结果可以看出,c和d除了数据类型不同导致的精度不一样之外,结果是相同的。