torch.linalg¶
常见的线性代数运算。
有关一些常见的数值边缘情况,请参见线性代数 (torch.linalg)。
矩阵属性¶
计算向量或矩阵范数。 |
|
计算向量范数。 |
|
计算矩阵范数。 |
|
计算方阵的行列式。 |
|
计算方阵行列式的绝对值的符号和自然对数。 |
|
计算矩阵相对于矩阵范数的条件数。 |
|
计算矩阵的数值秩。 |
分解¶
计算复 Hermitian 矩阵或实对称正定矩阵的 Cholesky 分解。 |
|
计算矩阵的 QR 分解。 |
|
计算矩阵部分透视的 LU 分解。 |
|
计算 LU 因式分解的紧凑表示形式,其中包含矩阵的部分透视。 |
|
计算方阵的特征值分解(如果存在)。 |
|
计算方阵的特征值。 |
|
计算复数 Hermitian 或实对称矩阵的特征值分解。 |
|
计算复 Hermitian 矩阵或实对称矩阵的特征值。 |
|
计算矩阵的奇异值分解 (SVD)。 |
|
计算矩阵的奇异值。 |
解决¶
计算具有唯一解的线性方程的平方组的解。 |
|
计算具有唯一解的线性方程三角方程组的解。 |
|
计算线性方程的平方组的解,其中具有给定 LU 分解的唯一解。 |
|
计算线性方程组的最小二乘问题的解。 |
矩阵函数¶
计算方阵的矩阵指数。 |
|
计算整数 n 的方阵的 n 次方。 |
Matrix 产品¶
计算两个 3 维向量的叉积。 |
|
计算沿一个维度的两批向量的点积。 |
|
通过对乘法进行重新排序,以便执行最少的算术运算,从而有效地将两个或多个矩阵相乘。 |
|
计算 Householder 矩阵乘积的前 n 列。 |
实验函数¶
计算复 Hermitian 矩阵或实对称正定矩阵的 Cholesky 分解。 |
|
计算方阵的逆矩阵(如果它是可逆的)。 |
|
计算 Hermitian 矩阵或对称(可能是不定)矩阵的 LDL 因式分解的紧凑表示。 |
|
使用 LDL 因式分解计算线性方程组的解。 |