torch.linalg.det

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

计算方形矩阵的行列式。

支持浮点型、双精度型、复数浮点型和复数双精度型的数据类型作为输入。还支持矩阵的批量处理,如果A是一组矩阵,那么输出将具有相同的批处理维度。

参见

torch.linalg.slogdet() 计算方阵行列式的符号及其绝对值的自然对数。

参数

A (Tensor) – 形状为(*, n, n)的张量,其中*表示零个或多个批次维度。

关键字参数

out (Tensor, optional) – 输出张量。默认为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])
本页目录