本文共 571 字，大约阅读时间需要 1 分钟。

import torchimport numpy as npx = torch.empty(5,3)print(x)#初始化一个随机矩阵x = torch.zeros(5, 3, dtype=torch.long)print(x)#构建一个全0矩阵x=x.new_ones(5, 3, dtype=torch.double)x=torch.randn_like(x, dtype=torch.float)print(x)#与numpy的操作差不多print(x.size())#基本计算如下y=torch.rand(5,3)print(x+y)print(torch.add(x,y))#索引 :表示取所有 1表示列print(x[:,1])#view操作可以改变矩阵的维度x=torch.randn(4,4)  #4*4y=x.view(16)        #view 拉长z=x.view(-1,8)      #-1表示自动计算 8代表 ？*8 ？自动计算->2*8print(x.size(),y.size(),z.size())#与numpy协同操作a = torch.ones(5)b= a.numpy()print(b)c = np.ones(5)d = torch.from_numpy(c) #tensorprint(d)

转载地址：http://pxtgn.baihongyu.com/