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格式
Pytorch——基本操作、与numpy协同
猜你喜欢
转载自blog.csdn.net/harryxia2014/article/details/122675408
今日推荐
周排行