squeeze（）

最新推荐文章于 2025-06-30 10:32:15 发布

O_o381

最新推荐文章于 2025-06-30 10:32:15 发布

阅读量490

点赞数 2

CC 4.0 BY-SA版权

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_61706514/article/details/143862919

squeeze 函数是 PyTorch 中的一个函数，用于从张量（Tensor）中去除所有长度为 1 的维度。这在处理神经网络模型的输出时非常有用，因为模型的输出可能包含一些不必要的单一维度，例如在批处理大小为 1 的情况下，输出可能会有一个额外的批次维度。

简单来说就是从张量（Tensor）中去除所有长度为 1 的维度。

实例：

import torch

x = torch.randn(1, 2, 1, 2, 2)
print(x.shape)  # 输出: torch.Size([1, 2, 1, 2, 2])

y = torch.squeeze(x)
print(y.shape)  # 输出: torch.Size([2, 2, 2])

z = torch.squeeze(x, dim=1)
print(z.shape)  # 输出: torch.Size([1, 2, 1, 2, 2])

v= torch.squeeze(x, dim=0)
print(v.shape)  # 输出: torch.Size([1, 2, 2, 2])

输出：

#x：

torch.Size([1, 2, 1, 2, 2])

#y：不指定维度就去除所有长度为 1 的维度
torch.Size([2, 2, 2])

#z：指定维度长度不为1则无效
torch.Size([1, 2, 1, 2, 2])

#v：指定维度长度为1生效
torch.Size([2, 1, 2, 2])