01-pytorch

import torch
import numpy as np

np和torch互相转换

np_data = np.arange(6).reshape((2,3))
#np ---> torch
torch_data = torch.from_numpy(np_data)
#torch --> np
torch_data_2= torch_data.numpy()

[[0 1 2]
[3 4 5]]
tensor([[0, 1, 2],
[3, 4, 5]])

torch的运算

求绝对值
torch.abs(tensor)
矩阵向乘
torch.mm(tensor,tensor)

data = [-1,-2,1,2]
tensor = torch.FloatTensor(data)
torch.abs(tensor)