html在网站开发中的应用,奥远网站建设流程,eclipse网站开发例子,企业定制在深度学习中经常会遇见这个函数#xff0c;现在来说一下这个函数的用法#xff0c;其实很简单的。
函数作用
函数的作用就是#xff1a;挤压size为1的维度#xff0c;挤压也就是remove。如果size不是1的话#xff0c;那就没变化。
使用说明
在使用的时候#xff0c;…在深度学习中经常会遇见这个函数现在来说一下这个函数的用法其实很简单的。
函数作用
函数的作用就是挤压size为1的维度挤压也就是remove。如果size不是1的话那就没变化。
使用说明
在使用的时候参数有torch.squeeze(input, dimNone)
至于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])
