torch.linalg¶
常见的线性代数运算。
参见 线性代数 (torch.linalg) 了解一些常见的数值边缘情况。
矩阵属性¶
计算向量或矩阵范数。 |
|
计算向量范数。 |
|
计算矩阵范数。 |
|
等效于 |
|
计算方阵的行列式。 |
|
计算方阵的行列式的符号和自然对数。 |
|
计算矩阵相对于矩阵范数的条件数。 |
|
计算矩阵的数值秩。 |
分解¶
计算复厄米特或实对称正定矩阵的 Cholesky 分解。 |
|
计算矩阵的 QR 分解。 |
|
计算矩阵的 LU 分解(带部分主元) 。 |
|
计算矩阵的 LU 分解(带部分主元)的紧凑表示。 |
|
如果存在,计算方阵的特征值分解。 |
|
计算方阵的特征值。 |
|
计算复厄米特或实对称矩阵的特征值分解。 |
|
计算复厄米特或实对称矩阵的特征值。 |
|
计算矩阵的奇异值分解 (SVD)。 |
|
计算矩阵的奇异值。 |
求解器¶
计算具有唯一解的方程组的解。 |
|
计算具有唯一解的三角方程组的解。 |
|
给定 LU 分解,计算具有唯一解的方程组的解。 |
|
计算线性方程组的最小二乘问题的解。 |
矩阵函数¶
计算方阵的矩阵指数。 |
|
计算方阵的 n 次方,其中 n 是整数。 |
矩阵乘积¶
计算两个三维向量的叉积。 |
|
等效于 |
|
计算沿某个维度上的两个向量批次的点积。 |
|
通过重新排序乘法来有效地乘以两个或多个矩阵,从而执行最少的算术运算。 |
|
计算 Householder 矩阵乘积的前 n 列。 |
张量运算¶
计算 |
|
计算系统 torch.tensordot(A, X) = B 的解 X。 |
实验性函数¶
计算复厄米特或实对称正定矩阵的 Cholesky 分解。 |
|
如果方阵可逆,则计算其逆。 |
|
|
|
这是 |
|
计算厄米特或对称(可能是不定)矩阵的 LDL 分解的紧凑表示。 |
|
这是 |
|
使用 LDL 分解计算线性方程组的解。 |