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

相关推荐
之歆6 小时前
Spring AI入门到实战到原理源码-MCP
java·人工智能·spring
知乎的哥廷根数学学派6 小时前
面向可信机械故障诊断的自适应置信度惩罚深度校准算法(Pytorch)
人工智能·pytorch·python·深度学习·算法·机器学习·矩阵
且去填词6 小时前
DeepSeek :基于 Schema 推理与自愈机制的智能 ETL
数据仓库·人工智能·python·语言模型·etl·schema·deepseek
待续3017 小时前
订阅了 Qoder 之后,我想通过这篇文章分享一些个人使用心得和感受。
人工智能
weixin_397578027 小时前
人工智能发展历史
人工智能
人工干智能7 小时前
OpenAI Assistants API 中 client.beta.threads.messages.create方法,兼谈一星*和两星**解包
python·llm
databook7 小时前
当条形图遇上极坐标:径向与圆形条形图的视觉革命
python·数据分析·数据可视化
强盛小灵通专卖员7 小时前
基于深度学习的山体滑坡检测科研辅导:从论文实验到系统落地的完整思路
人工智能·深度学习·sci·小论文·山体滑坡
OidEncoder7 小时前
从 “粗放清扫” 到 “毫米级作业”,编码器重塑环卫机器人新能力
人工智能·自动化·智慧城市
Hcoco_me7 小时前
大模型面试题61:Flash Attention中online softmax(在线softmax)的实现方式
人工智能·深度学习·自然语言处理·transformer·vllm