torch.linalg

常見的線性代數運算。

請參閱線性代數 (torch.linalg) 以取得一些常見的數值邊緣情況。

矩陣屬性

norm	計算向量或矩陣範數。
vector_norm	計算向量範數。
matrix_norm	計算矩陣範數。
diagonal	是 torch.diagonal() 的別名，預設值為 dim1= -2, dim2= -1。
det	計算方陣的行列式。
slogdet	計算方陣行列式絕對值的符號和自然對數。
cond	計算矩陣相對於矩陣範數的條件數。
matrix_rank	計算矩陣的數值秩。

分解

cholesky	計算複 Hermitian 或實對稱正定矩陣的 Cholesky 分解。
qr	計算矩陣的 QR 分解。
lu	計算矩陣的部分樞軸 LU 分解。
lu_factor	計算矩陣的部分樞軸 LU 分解的簡潔表示。
eig	計算方陣的特徵值分解（如果存在）。
eigvals	計算方陣的特徵值。
eigh	計算複 Hermitian 或實對稱矩陣的特徵值分解。
eigvalsh	計算複 Hermitian 或實對稱矩陣的特徵值。
svd	計算矩陣的奇異值分解 (SVD)。
svdvals	計算矩陣的奇異值。

求解器

solve	計算具有唯一解的線性方程組的解。
solve_triangular	計算具有唯一解的三角線性方程組的解。
lu_solve	計算給定 LU 分解的線性方程組的解。
lstsq	計算線性方程組的最小二乘問題的解。

反矩陣

inv	計算方陣的反矩陣（如果存在）。
pinv	計算矩陣的偽逆（Moore-Penrose 逆矩陣）。

矩陣函數

matrix_exp	計算方陣的矩陣指數。
matrix_power	對於整數 n，計算方陣的 n 次方。

矩陣乘積

cross	計算兩個 3 維向量的叉積。
matmul	是 torch.matmul() 的別名
vecdot	沿一個維度計算兩批向量的點積。
multi_dot	通過重新排序乘法來有效率地乘兩個或多個矩陣，以便執行最少的算術運算。
householder_product	計算 Householder 矩陣乘積的前 n 列。

張量運算

tensorinv	計算 torch.tensordot() 的乘法逆矩陣。
tensorsolve	計算系統 torch.tensordot(A, X) = B 的解 X。

其他

vander

產生 Vandermonde 矩陣。

實驗性函數

cholesky_ex	計算複 Hermitian 或實對稱正定矩陣的 Cholesky 分解。
inv_ex	計算方陣的反矩陣（如果可逆）。
solve_ex	是 solve() 的一個版本，除非 check_errors= True，否則不執行錯誤檢查。
lu_factor_ex	這是 lu_factor() 的一個版本，除非 check_errors= True，否則不執行錯誤檢查。
ldl_factor	計算 Hermitian 或對稱（可能是不定的）矩陣的 LDL 分解的簡潔表示。
ldl_factor_ex	這是 ldl_factor() 的一個版本，除非 check_errors= True，否則不執行錯誤檢查。
ldl_solve	使用 LDL 分解計算線性方程組的解。