当前位置: 首页> 教育> 高考 > torch.squeeze()

torch.squeeze()

时间:2025/7/9 6:51:26来源:https://blog.csdn.net/qq_43701910/article/details/142101201 浏览次数:1次

在深度学习中经常会遇见这个函数,现在来说一下这个函数的用法,其实很简单的。

函数作用

函数的作用就是:挤压size为1的维度,挤压也就是remove。如果size不是1的话,那就没变化。

使用说明

在使用的时候,参数有:torch.squeeze(input, dim=None)

  • 至于dim:可以传入一个维度或者是多个维度的元祖形式
举例
>>> x = torch.zeros(2, 1, 2, 1, 2)
>>> x.size()
torch.Size([2, 1, 2, 1, 2])#仅仅传入tensor,不指定具体维度的话,默认是对所有的维度进行挤压
>>> y = torch.squeeze(x)
>>> y.size()
torch.Size([2, 2, 2])# 指定对维度0 进行挤压,但是维度0的size是2,所以没变化
>>> y = torch.squeeze(x, 0)
>>> y.size()
torch.Size([2, 1, 2, 1, 2])# 指定对维度1进行挤压,因为维度1的size是1,因此进行加压后变减少了一个维度
>>> y = torch.squeeze(x, 1)
>>> y.size()
torch.Size([2, 2, 1, 2])# 传入多个维度通过元祖的形式,
>>> y = torch.squeeze(x, (1, 2, 3))
torch.Size([2, 2, 2])
关键字:torch.squeeze()

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: