Pytorch当中squeeze(1)函数和unsqueeze(1)函数的区别及含义

一、squeeze(1)

在 PyTorch 中,.squeeze(1) 是用于张量维度操作的方法之一,它的作用是将张量中尺寸为 1 的维度压缩(去除)掉。

具体来说,如果张量在指定维度(这里是维度 1)上的尺寸为 1,.squeeze(1) 方法将会移除这个维度,从而减少张量的维度数。

举例说明,假设有一个形状为 (A, 1, B, C) 的张量,其中维度 1 的尺寸为 1。使用 .squeeze(1) 操作后,将会得到一个新的张量,其形状为 (A, B, C),维度 1 被压缩掉了。

需要注意的是,.squeeze() 方法默认会压缩所有尺寸为 1 的维度,如果指定了参数(例如 .squeeze(1)),则只会在指定的维度上进行压缩。这种操作通常用于消除尺寸为 1 的维度,以便更好地与其他张量进行操作或匹配。

二、unsqueeze(1)

在 PyTorch 中,.unsqueeze(1) 是用于张量维度操作的方法之一,它的作用是在指定位置(这里是维度 1)上增加一个维度,将维度的大小设置为 1。

具体来说,.unsqueeze(1) 方法会在指定的位置(这里是维度 1)上增加一个新的维度,使得张量的维度数增加,并将新增的维度的大小设置为 1。

举例来说,如果有一个形状为 (A, B, C) 的张量,在维度 1 上使用 .unsqueeze(1) 操作后,将会得到一个新的形状为 (A, 1, B, C) 的张量。这表示在原来的张量中的维度 1 处增加了一个维度,该维度的大小为 1。

.unsqueeze() 方法用于在指定位置增加维度,通常在需要对张量进行扩展或与其他维度不匹配的情况下使用。

相关推荐
Channing Lewis19 分钟前
pyproject.toml
python
测试人社区—667921 分钟前
破茧成蝶:DevOps流水线测试环节的效能跃迁之路
运维·人工智能·学习·flutter·ui·自动化·devops
Lethehong1 小时前
openEuler AI 图像处理:Stable Diffusion CPU 推理性能优化与评测
人工智能
Guheyunyi1 小时前
智慧停车管理系统:以科技重塑交通效率与体验
大数据·服务器·人工智能·科技·安全·生活
std860211 小时前
微软将允许用户从Windows 11文件资源管理器中移除“AI 动作”入口
人工智能·microsoft
为爱停留1 小时前
Spring AI实现MCP(Model Context Protocol)详解与实践
java·人工智能·spring
秋刀鱼 ..1 小时前
第七届国际科技创新学术交流大会暨机械工程与自动化国际学术会议(MEA 2025)
运维·人工智能·python·科技·机器人·自动化
xwill*8 小时前
分词器(Tokenizer)-sentencepiece(把训练语料中的字符自动组合成一个最优的子词(subword)集合。)
开发语言·pytorch·python
学历真的很重要8 小时前
VsCode+Roo Code+Gemini 2.5 Pro+Gemini Balance AI辅助编程环境搭建(理论上通过多个Api Key负载均衡达到无限免费Gemini 2.5 Pro)
前端·人工智能·vscode·后端·语言模型·负载均衡·ai编程
普通网友8 小时前
微服务注册中心与负载均衡实战精要,微软 2025 年 8 月更新:对固态硬盘与电脑功能有哪些潜在的影响。
人工智能·ai智能体·技术问答