快捷方式

torch.triangular_solve

torch.triangular_solve(b, A, upper=True, transpose=False, unitriangular=False, *, out=None)

求解具有方形上三角或下三角可逆矩阵 AA 和多个右手边 bb 的方程组。

用符号表示,它求解 AX=bAX = b 并假设 AA 是方形上三角(如果 upper= False 则为下三角)并且对角线上没有零。

torch.triangular_solve(b, A) 可以接收 2D 输入 b, A 或 2D 矩阵的批处理输入。如果输入是批处理,则返回批处理输出 X

如果 A 的对角线包含零或非常接近零的元素,并且 unitriangular= False(默认)或如果输入矩阵条件很差,则结果可能包含 NaN

支持 float、double、cfloat 和 cdouble 数据类型的输入。

警告

torch.triangular_solve() 已被弃用,取而代之的是 torch.linalg.solve_triangular(),并且将在未来的 PyTorch 版本中删除。 torch.linalg.solve_triangular() 的参数顺序相反,并且不会返回其中一个输入的副本。

X = torch.triangular_solve(B, A).solution 应替换为

X = torch.linalg.solve_triangular(A, B)
参数
  • b (Tensor) – 大小为 (,m,k)(*, m, k) 的多个右手边,其中 * 是零个或多个批处理维度

  • A (张量) – 输入三角系数矩阵,大小为 (,m,m)(*, m, m),其中 * 表示零个或多个批次维度

  • upper (布尔值, 可选) – AA 是上三角还是下三角。默认值:True

  • transpose (布尔值, 可选) – 求解 op(A)X = b,其中 op(A) = A^T 如果此标志为 True,并且 op(A) = A 如果此标志为 False。默认值:False

  • unitriangular (布尔值, 可选) – AA 是单位三角形。如果为 True,则 AA 的对角线元素被假定为 1,并且不会从 AA 中引用。默认值:False

关键字参数

out ((张量, 张量), 可选) – 用于写入输出的两个张量的元组。如果为 None 则忽略。默认值:None.

返回值

一个名为元组 (solution, cloned_coefficient),其中 cloned_coefficientAA 的克隆,而 solutionAX=bAX = b(或根据关键字参数的不同,系统方程的任何变体)的解 XX

示例

>>> A = torch.randn(2, 2).triu()
>>> A
tensor([[ 1.1527, -1.0753],
        [ 0.0000,  0.7986]])
>>> b = torch.randn(2, 3)
>>> b
tensor([[-0.0210,  2.3513, -1.5492],
        [ 1.5429,  0.7403, -1.0243]])
>>> torch.triangular_solve(b, A)
torch.return_types.triangular_solve(
solution=tensor([[ 1.7841,  2.9046, -2.5405],
        [ 1.9320,  0.9270, -1.2826]]),
cloned_coefficient=tensor([[ 1.1527, -1.0753],
        [ 0.0000,  0.7986]]))

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源