【pytorch算子】torch.ger / torch.outer算子的用法,以及与torch库版本的关系

torch.ger 的用法

torch.ger 用于实现两个 Tensor 的内积,即 ger(a,b)= a^T *b (其中 a^T 表示 a 的转置)

example:

a=torch.tensor([2,3,4,5])
b=torch.tensor([1,2,3])
c=torch.ger(a,b)
print(c)

输出c:

tensor([[ 2,  4,  6],
        [ 3,  6,  9],
        [ 4,  8, 12],
        [ 5, 10, 15]])

a 是 14 的 Tensor,转置后形状为 41 ;b 是 13 的 Tensor,因此 c = ger(a,b) 是 43 的 Tensor

pytorch版本

查看 pytorch 官方文档可以看到, torch.ger 算子即将被删除,用 torch.outer 代替:

torch

程序运行报错:显示torch没有outer算子

在这里插入图片描述
查看torch版本发现我的torch版本是1.5.0

在这里插入图片描述
关键点来了:

在pytorch1.7.0之后,torch.ger算子被删掉了,用torch.outer代替。

也就是说,只有pytorch1.7.0的torch.outer与torch.ger功能等价,因此torch1.5.0没有outer算子。

于是我安装了最新的torch1.8.1,当然原来的torch1.5.0也自动被uninstalled了

在这里插入图片描述
安装好之后,再次运行程序,没有报错,问题解决。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ctrl A_ctrl C_ctrl V

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值