torch.linalg

常見的線性代數運算。

有關常見數值邊緣情況，請參閱線性代數 (torch.linalg)。

矩陣屬性

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

分解

cholesky	計算復厄米特矩陣或實對稱正定矩陣的 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 次冪，其中 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	計算復厄米特矩陣或實對稱正定矩陣的 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 因式分解計算線性方程組的解。