pytorch中squeeze()和unsqueeze()
squeeze的用法主要就是对数据的维度进行压缩或者扩充。torch.squeeze() 对数据的维度进行压缩。去掉维数为1的的维度,squeeze(a)将a中所有为1的维度删掉,不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=torch.squeeze(a,N) ,在a中指定位置N去掉一个维数为1的维度torch.unsqu...
原创
2019-06-10 12:16:51 ·
5169 阅读 ·
0 评论