#【PyTorch】squeeze函数详情
函数原型torch.squeeze(input,dim=None,*,out=None ) Tensor函数详细信息删除一个ssdsy中所有维数为1的维度
例如,(A 1 B C 1 D )维度) a )时间1 )时间)时间)时间) A 1 B C 1 D )的ssdsy调用ssy
如果指定dim参数,则squeeze操作仅应用于指定的维。 输入input的形状为(A 1 B ) a ) times1) A times B ) A 1 B )时,squeeze ) input,0 )不变更该ssdsy,但squeeze ) input,
示例x=torch.zeros (2,1,2,1,2 ) x.size ) Torch.size ) [ 2,1,2,1,2 ] ) y=torch.squeze ) x ) y.size