pytorch学习（一）

tensor张量

tensor能够用很多种方法初始化

import torch
import numpy as np
data=[[1,2],[3,4]]
x_data=torch.tensor(data)

np_array=np.array(data)
x_np=torch.from_numpy(np_array)

x_ones=torch.ones_like(x_data)
x_rand=torch.rand_like(x_data,dtype=torch.float)

rand_tensor=torch.rand(shape)
ones_tensor=torch.ones(shape)
zeros_tensor=torch.zeros(shape)

tensor属性及操作

算法

two=torch.ones((2,2))
two=two.to_cuda()
torch.matmul(tensor, two)
two @ tensor
agg=tensor.sum()
agg.item()