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])