记录squeeze()

最新推荐文章于 2024-06-03 21:49:40 发布

懒羊羊001

最新推荐文章于 2024-06-03 21:49:40 发布

阅读量986

点赞数

文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/freedomsm/article/details/122228900

版权

这篇博客介绍了 PyTorch 中的 squeeze() 函数，该函数用于移除张量中维度为 1 的轴。通过示例代码展示了如何将一个形状为 torch.size([2,3,2,1]) 的张量转换为 torch.size([2,3,2])。文章通过实际操作解释了 squeeze() 在张量操作中的应用。

摘要由CSDN通过智能技术生成

'''
squeeze()函数可以去除维度为1的维度，比如a的形状为torch.size([2,3,2,1]),使用squeeze()函数后就会变为
torch.size([2,3,2])
import torch
ysh=torch.arange(0,10)
ysh=ysh.view(2,5)
mess=torch.arange(0,16).view(2,4,2,1)
print('mess的形状',mess.shape)
print('初次形状',ysh.shape)
ysh=torch.squeeze(mess)
print("二次形状",ysh.shape)
print(ysh)
'''