解决问题:ValueError: only one element tensors can be converted to Python scalars

问题代码:

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除了数据类型不同导致的精度不一样之外,结果是相同的。

你可能感兴趣的:(error,I,met)