pytorch学习(一)squeeze()与unsqueeze()
1、squeeze()
顾名思义,这个函数主要对数据的维度进行压缩。
这个函数对用来压缩维度为1的数据。不管是行数据或者是列数据。总之就是一把梭。这个时候我们来看下列代码。
import torch
a=torch.randn(1,3,4,5)
print(a)
输出的结果:
tensor([[[[-0.8684, 0.0383, 1.5879, -0.6658, -0.3179],
[ 1.3566, 0.0889, -0.9394, 0.1673, -1.1042],
[-0.8013,
原创
2020-06-04 22:43:49 ·
596 阅读 ·
0 评论