Pytorch Tensor维度变换方法

1.torch.reshape()、torch.view()可以调整Tensor的shape

2.torch.unsqueeze(index)可以为Tensor增加一个维度

3.squeeze()可以删减维度

4.expand()扩展维度

5.repeat()维度重复,不常用

6 .transpose(dim1, dim2)交换dim1与dim2,进行维度变换。

注意这种交换使得存储不再连续,再执行一些reshape的操作会报错,所以要调用一下contiguous()使其变成连续的维度。

参考:https://blog.csdn.net/weicao1990/article/details/93618136

7.Broadcasting(广播)也就是自动实现了若干unsqueeze和expand操作,以使两个Tensor的shape一致,从而完成某些操作(往往是加法)。

比如得到的feature maps的维度是4,32,14,14,对这些特征图加上偏置,bias32,先对bias进行unsqueeze操作增加维度:1,32,1,1,再用expand操作进行扩充4,32,14,14。然后两者就可以相加了。

相关推荐
AI导出鸭2 小时前
Claude的LaTeX生成PDF文件复制后数学公式乱码,怎样修改?专业用户首选“AI导出鸭”
人工智能·pdf·ai导出鸭
三十岁老牛再出发2 小时前
8月21日总结
c++·python
染指11102 小时前
95.RAG-RAG应用平台-工作流(Agent)
人工智能·agent
草莓熊Lotso3 小时前
【Linux网络加餐】手动部署:SSH 与 Web 服务实战 + 底层原理全解析
linux·运维·网络·人工智能·python·langchain·ssh
小妖同学学AI3 小时前
60.8k星!开源金融数据神器OpenBB:连接一切数据,让分析师、量化交易员和AI智能体如虎添翼!
人工智能·金融·开源
weixin_440730503 小时前
python+request实现接口-参数化小结
开发语言·python
YOLO数据集集合3 小时前
UAVDT 无人机车辆检测数据集 - 无人机航拍 | 车辆检测 | 目标检测 | YOLO格式 | 智能交通 | 城市管理 | 多类别数据集 | 计算机视觉
人工智能·yolo·目标检测·机器学习·计算机视觉·目标跟踪·无人机
aneasystone本尊3 小时前
大模型推理介绍:从一次提问说起
人工智能
小王2043 小时前
Day 21:卷积神经网络CNN — 深入理解卷积
人工智能·神经网络·cnn
DevNo3 小时前
学生党课堂复习利器:三款音视频转文字工具实测对比
人工智能