torch.linalg

創建於：2020年8月7日 | 最後更新於：2025年6月17日

常用的線性代數運算。

有關一些常見的數值邊緣情況，請參閱 線性代數 (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	計算複共軛厄米矩陣或實對稱矩陣的特徵值分解。
eigvalsh	計算複共軛厄米矩陣或實對稱矩陣的特徵值。
svd	計算矩陣的奇異值分解 (SVD)。
svdvals	計算矩陣的奇異值。

求解器

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

逆

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

矩陣函式

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

矩陣乘積

cross	計算兩個三維向量的叉積。
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	計算厄米或對稱（可能不確定）矩陣的 LDL 分解的緊湊表示。
ldl_factor_ex	是 ldl_factor() 的一個版本，除非 check_errors= True，否則不執行錯誤檢查。
ldl_solve	使用 LDL 分解計算線性方程組的解。