pytorch学习强推莫烦up主:https://github.com/MorvanZhou/PyTorch-Tutorial
1. numpy与tensor
x = torch.Tensor(5, 3) #创建一个未初始化的5*3矩阵
x = torch.zeros(5,3, dtype=torch.long)#创建一个数据类型为long的零矩阵
x = torch.Tensor([5.5, 3]) #用数据创造一个张量
#numpy bridge以 numpy作媒介 实现tensor与numpy的转化
#Torch Tensor 和 Numpy 数组共享底层的内存位置,改变任何一个都将改变另一个.
np_data = np.arange(6).reshape((2,3)) #创造numpy数据
torch_data = torch.from_numpy(np_data) #转化为torch数据
tensor2array = torch_data.numpy() #torch数据转化为numpy数据
print(
'\nnumpy', np_data,
'\ntorch', torch_data,
'\ntensor2array', tensor2array,
)
#结果
numpy [[0 1 2]
[3 4 5]]
torch tensor([[0, 1, 2],
[3, 4, 5]], dtype=torch.int32)
tensor2array [[0 1 2]
[3 4 5]]
#矩阵相乘,numpy两种方法,torch为一种
data = [[1,2]