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

相关推荐
要吃这碗饭1 分钟前
AI 智能体如何通过 auth.md 注册 Bright Data 账号
数据库·人工智能·php
天远Date Lab10 分钟前
零信任架构实战:基于天远学籍核验三要素构建自动化竞赛资格审查网关
运维·人工智能·架构·自动化
海盗123411 分钟前
AI 新闻日报 2026-10-09:企业代理独立身份、便利店机器人上岗、世界模型接棒
人工智能·机器人·人工智能aigc
核数聚15 分钟前
低端标注拼的是单价,高端标注拼的是能力
人工智能·核数聚
东离与糖宝21 分钟前
RAG调参的承重项:Agentic AutoRAG背后藏着反转
人工智能
FL162386312925 分钟前
外来入侵植物检测数据集VOC+YOLO格式1111张5类别
人工智能·yolo·机器学习
oioihoii31 分钟前
三步改掉 AI 味,附可直接复制的去 AI 味提示词
人工智能
阿乔外贸日记32 分钟前
全球电子产业高度集中:中、韩、日和中国台湾地区占全球数字硬件产能八成
大数据·服务器·人工智能·物联网·云计算
曹牧38 分钟前
AI Skill
人工智能
2601_9628857242 分钟前
如何用 Python 计算 BBI 多空指标?
开发语言·python