PyTorch torch.linalg.det 函数
Pytorch torch 参考手册torch.linalg.det 是 PyTorch 线性代数模块中用于计算矩阵行列式的函数。它是 torch.det 的推荐替代函数。
函数定义
torch.linalg.det(A)
参数:
A(Tensor): 输入方阵。
返回值:
torch.Tensor: 返回矩阵的行列式。
使用示例
实例
import torch
# 创建方阵
A = torch.tensor([[1.0, 2.0, 3.0],
[0.0, 4.0, 5.0],
[0.0, 0.0, 6.0]])
# 计算行列式
det_A = torch.linalg.det(A)
print("矩阵 A:")
print(A)
print("行列式:")
print(det_A)
# 创建方阵
A = torch.tensor([[1.0, 2.0, 3.0],
[0.0, 4.0, 5.0],
[0.0, 0.0, 6.0]])
# 计算行列式
det_A = torch.linalg.det(A)
print("矩阵 A:")
print(A)
print("行列式:")
print(det_A)
输出结果为：
矩阵 A:
tensor([[1., 2., 3.],
[0., 4., 5.],
[0., 0., 6.]])
行列式:
tensor(24.)