Torch和Numpy——求行列式

输入

import numpy as np
import torch

#numpy求法
a = np.array([[1,4,7],[2,5,8,],[3,6,0]])
print(np.linalg.det(a))

#torch求法
b = torch.tensor([[1,4,7],[2,5,8],[3,6,0]],dtype=torch.float32)
print(torch.det(b))

输出

27.0
tensor(27.)