評價此頁

torch.linalg.det#

torch.linalg.det(A, *, out=None) Tensor#

計算方陣的行列式。

支援浮點 (float)、雙精度浮點 (double)、複數浮點 (cfloat) 和複數雙精度浮點 (cdouble) 資料型別。還支援矩陣批處理,如果 `A` 是一個矩陣批處理,則輸出具有相同的批處理維度。

另請參閱

torch.linalg.slogdet() 計算方陣行列式符號和絕對值自然對數。

引數

A (Tensor) – 形狀為 (*, n, n) 的張量,其中 * 是零個或多個批次維度。

關鍵字引數

out (Tensor, optional) – 輸出張量。如果為 None 則忽略。預設為 None

示例

>>> A = torch.randn(3, 3)
>>> torch.linalg.det(A)
tensor(0.0934)

>>> A = torch.randn(3, 2, 2)
>>> torch.linalg.det(A)
tensor([1.1990, 0.4099, 0.7386])