Pytorch——基本操作、与numpy协同

import torch
import numpy as np

x = torch.empty(5,3)#创建5行3列的0矩阵

y = torch.rand(5,3)#创建5行3列的随机数矩阵

a = torch.empty(5,3,dtype=torch.double)#创建5行3列的0矩阵

x.size()#打印维度

x + y#基本计算一会

b = torch.empty(4,4)

c = b.view(16)#改变形状

d = b.view(-1,8)#改变形状

e = b.numpy()#转成numpy格式

f = np.ones(5)

g = torch.from_numpy(f)#转成PyTorch格式