torch.solve已弃用,如何改为torch.linalg.solve?

48 篇文章 1 订阅
10 篇文章 0 订阅
本文介绍了在PyTorch中,`torch.solve`函数被弃用,推荐使用`torch.linalg.solve`来解决线性方程组。两者输入顺序不同,torch.linalg.solve仅返回解向量,不包含LU分解信息。开发者应更新代码以适应新函数并获取完整解信息。
摘要由CSDN通过智能技术生成

具体报错:RuntimeError: This function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.solve` function instead.

报错代码:

X, LU = torch.solve(rhs, lhs)

1、torch.solve 函数的期望输入有2个:右侧向量、系数矩阵;而 torch.linalg.solve 函数的期望输入也有2个:系数矩阵、右侧向量(顺序与torch.solve的输入相反)。

2、torch.solve 和 torch.linalg.solve 在返回值上也有所不同。

torch.solve 函数返回一个元组 (solution, LU),其中 solution 是解向量,LU 是矩阵 A 的 LU 分解。LU 分解是一种将矩阵分解为下三角矩阵 L 和上三角矩阵 U 的方法。

而 torch.linalg.solve 函数只返回解向量,不返回 LU 分解。因此,torch.linalg.solve 只给出了线性方程组的解,而没有提供 LU 分解的信息。

所以这行代码需要改为:

X = torch.linalg.solve(lhs, rhs)

  • 15
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

我有明珠一颗

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

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

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

打赏作者

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

抵扣说明:

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

余额充值