pytorch和numpy的互相转换、数学计算公式

格式转换

使用torch.from_numpy() 将numpy的array转化为torch的张量。

结果如下： 

 

而把张量转为numpy的数组：tensor2array = torch_data.numpy()

---

计算绝对值 torch.abs()

data = [-1,-2，1，2]
tensor = torch.FloatTensor(data)    #32bit

print(torch.abs(tensor))

---

torch的 矩阵乘法  （dot函数/mm函数）

矩阵相乘应该使用：torch.mm(tensor_a,tensor_b)  或者  torch.matmul(tensor_a,tensor_b)

import torch
a=torch.randn(2,3)
b=torch.randn(3,2)
print(torch.mm(a,b))
# 或者 print(torch.matmul(a,b))

注意：torch.dot()和 np.dot()是完全不一样的！

np.dot()也是矩阵相乘的函数

而torch.dot()则是把张量展开成一维的，然后做乘法相加，得到的是一个数字。