Pytorch知识集锦（一） torch x=x.view(x.size()[0],-1)

在写特征提取网络的时候，在卷积网络和分类网络的相结合的地方，需要用到将特征转换成一个列数为一的一维向量：

#coding=utf-8
#torch x=x.view(x.size()[0],-1)
import torch
a=torch.arange(1,17)   #[1,16]
print(a)
print(a.size(0))   #[16]  一行，十六列
#x=x.view(x.size(0),-1)
b=a.view(a.size()[0],-1)
#b=a.view(a.size(0),-1)
print(b)#(16,1)   十六行，一列