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

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值