pytorch中squeeze函数用法

squeeze的中文意思是"挤压 ",顾名思义,该函数的作用是压缩维度

python 复制代码
squeeze(input, dim=None) -> Tensor

input一个高维张量,如果各个维度中存在大小为1的维度,squeeze才起作用,下面举例说明

python 复制代码
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,当dim=1时候,效果和上面相同,如果dim为其他,那么x的维度不变。

当然,如果各个维度中不存在大小为1的维度,那么squeeze函数对x张量无效。

相关推荐
Codebee几秒前
OneCode 3.0 从0到1干货——AIGC及MCP注解驱动开发物联网AI决策应用
人工智能·物联网
倔强青铜三1 分钟前
苦练Python第19天:断言与自定义异常
人工智能·python·面试
荷蒲4 分钟前
【小白量化智能体】应用5:编写通达信股票交易指标及生成QMT自动交易Python策略程序
开发语言·python
_一条咸鱼_11 分钟前
LangChain正则表达式(19)
人工智能·面试·langchain
墨绿色的摆渡人13 分钟前
具身智能零碎知识点(五):VAE中对使用KL散度的理解
人工智能·vae·具身智能
大模型真好玩13 分钟前
深入浅出LangChain AI Agent智能体开发教程(一)—认识LangChain&LangGraph
人工智能·python·mcp
汀丶人工智能17 分钟前
AI-Compass宝藏资源库:构建最全面的AI学习与实践生态,服务AI全群体
人工智能
阿里云大数据AI技术21 分钟前
NL2SQL 再创佳绩!阿里云论文中选 SIGMOD 2025
大数据·人工智能·云计算
小赖同学24 分钟前
CAMEL 框架之 RolePlaying 角色扮演
人工智能·后端·aigc
医工交叉实验工坊31 分钟前
R 语言绘制 10 种精美火山图:转录组差异基因可视化
python·信息可视化·r语言