在写特征提取网络的时候,在卷积网络和分类网络的相结合的地方,需要用到将特征转换成一个列数为一的一维向量:
#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) 十六行,一列