torch.squeeze

最新推荐文章于 2025-05-28 09:42:41 发布

weixin_30687811

最新推荐文章于 2025-05-28 09:42:41 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jiangkejie/p/10310299.html

本文详细介绍了PyTorch中squeeze函数的使用方法，包括如何移除张量中尺寸为1的维度，以及如何指定特定维度进行操作。通过实例展示了不同参数下squeeze函数的效果，强调了squeeze操作对内存的影响。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

squeeze可以将张量中尺寸为1的维度移除。

也可以指定维度。

注意：返回的张量与原张量共享内存，改变一个会同时改变另一个。

Example:

>>> x = torch.zeros(2, 1, 2, 1, 2)
>>> x.size()
torch.Size([2, 1, 2, 1, 2])
>>> y = torch.squeeze(x)
>>> y.size()
torch.Size([2, 2, 2])
>>> y = torch.squeeze(x, 0)
>>> y.size()
torch.Size([2, 1, 2, 1, 2])
>>> y = torch.squeeze(x, 1)
>>> y.size()
torch.Size([2, 2, 1, 2])