每天五分钟深度学习框架pytorch:tensor张量的维度转换大全

本文重点

在深度学习中比较让人头疼的一点就是矩阵的维度,我们必须构建出符合神经网络维度的矩阵,只有将符合要求的矩阵放到神经网络中才可以运行神经网络,本节课程我们将学习以下tensor中维度的变化。

view和shape

View和shape,这两个方法可以完成维度的变换操作,而且使用方法是一样的

压缩和解压张量squeeze和unsqueeze

压缩squeeze

Squeeze()如果不指定参数,则表示删除所有维度中为1的,如果指定参数,则删除指定索引中维度为1的,只操作维度为1的。

解压unsqueeze

解压就是在指定维度索引上插入维度1

上一篇:Pytorch实用教程:为什么定义模型时,没有输入形参,但是使用时可以直接传入输入数据?


下一篇:由标准库转HAL库(基于正点原子)——STM32HAL库常用函数——GPIO