pytorch中的torch.squeeze和torch.unsqueeze

python 复制代码
torch.squeeze(input, dim=None) → Tensor

torch.unsqueeze()函数的作用减少数组input指定的维度dim,如果dim不指定,则删除大小为1的维度。返回一个tensor。

如果数组A的维度为(1,1,3)那么执行 torch.squeeze(A,1) 后A的维度变为 (1,3)。

python 复制代码
torch.unsqueeze(input, dim) → Tensor

torch.unsqueeze()函数的作用增加数组input指定的维度dim。

例如两行三列的数组A维度为(2,3),那么这个数组就有三个位置可以增加维度,分别是( [位置0] 2,[位置1] 3 [位置2] )或者是 ( [位置-3] 2,[位置-2] 3 [位置-1] ),如果执行 torch.unsqueeze(A,1),数据的维度就变为了 (2,1,3)。

相关推荐
_BugMan23 分钟前
【大模型】理论基础(1):函数与神经网络
人工智能·深度学习·神经网络
hanfeng526828 分钟前
使用Deeplabv3+进行遥感影像土地利用分类
深度学习·神经网络·分类
java1234_小锋1 小时前
PyTorch2 Python深度学习 - 自动微分(Autograd)与梯度优化
开发语言·python·深度学习·pytorch2
AI模块工坊1 小时前
CVPR 即插即用 | PConv:重新定义高效卷积,一个让模型“跑”得更快、更省的新范式
人工智能·深度学习·计算机视觉·transformer
java1234_小锋1 小时前
PyTorch2 Python深度学习 - 简介以及入门
python·深度学习·pytorch2
lzjava20242 小时前
Spring AI加DeepSeek实现一个Prompt聊天机器人
人工智能·spring·prompt
fanstuck3 小时前
AI辅助数学建模有哪些优势?
人工智能·数学建模·语言模型·aigc
一只安3 小时前
从零开发AI(不依赖任何模型)
人工智能·python
11年老程序猿在线搬砖4 小时前
如何搭建自己的量化交易平台
大数据·人工智能·python·自动交易·量化交易系统
Elastic 中国社区官方博客4 小时前
Elasticsearch 开放推理 API 增加了对 Google 的 Gemini 模型的支持
大数据·人工智能·elasticsearch·搜索引擎·ai·全文检索·googlecloud