黄山网站建设公司,清镇市最新消息,销售网站建设,个人电脑做网站服务器网站squeeze的中文意思是“挤压”#xff0c;顾名思义#xff0c;该函数的作用是压缩维度
squeeze(input, dimNone) - Tensor
input一个高维张量#xff0c;如果各个维度中存在大小为1的维度#xff0c;squeeze才起作用#xff0c;下面举例说明
x torch.arange(6).res…squeeze的中文意思是“挤压”顾名思义该函数的作用是压缩维度
squeeze(input, dimNone) - Tensor
input一个高维张量如果各个维度中存在大小为1的维度squeeze才起作用下面举例说明
x torch.arange(6).reshape(2,1,3)# tensor([[[0, 1, 2]],
# [[3, 4, 5]]]) shape(2,1,3)x x.squeeze()# tensor([[0, 1, 2],
# [3, 4, 5]]) shape(2,3)
如果指定dim当dim1时候效果和上面相同如果dim为其他那么x的维度不变。
当然如果各个维度中不存在大小为1的维度那么squeeze函数对x张量无效。
