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张量无效。

相关推荐
周树皮不皮7 分钟前
20250704【翻转&二叉树】|Leetcodehot100之226【pass】&今天计划
python
魔芋红茶8 分钟前
spring-initializer
python·学习·spring
喜欢吃豆12 分钟前
快速手搓一个MCP服务指南(九): FastMCP 服务器组合技术:构建模块化AI应用的终极方案
服务器·人工智能·python·深度学习·大模型·github·fastmcp
星融元asterfusion19 分钟前
基于路径质量的AI负载均衡异常路径检测与恢复策略
人工智能·负载均衡·异常路径
一个天蝎座 白勺 程序猿19 分钟前
Python(28)Python循环语句指南:从语法糖到CPython字节码的底层探秘
开发语言·python
zskj_zhyl24 分钟前
智慧养老丨从依赖式养老到自主式养老:如何重构晚年生活新范式
大数据·人工智能·物联网
创小匠26 分钟前
创客匠人视角下创始人 IP 打造与知识变现的底层逻辑重构
人工智能·tcp/ip·重构
xiangduanjava41 分钟前
关于安装Ollama大语言模型本地部署工具
人工智能·语言模型·自然语言处理
zzywxc7871 小时前
AI 正在深度重构软件开发的底层逻辑和全生命周期,从技术演进、流程重构和未来趋势三个维度进行系统性分析
java·大数据·开发语言·人工智能·spring