本文重点
在深度学习中比较让人头疼的一点就是矩阵的维度,我们必须构建出符合神经网络维度的矩阵,只有将符合要求的矩阵放到神经网络中才可以运行神经网络,本节课程我们将学习以下tensor中维度的变化。
view和shape
View和shape,这两个方法可以完成维度的变换操作,而且使用方法是一样的
压缩和解压张量squeeze和unsqueeze
压缩squeeze
Squeeze()如果不指定参数,则表示删除所有维度中为1的,如果指定参数,则删除指定索引中维度为1的,只操作维度为1的。
解压unsqueeze
解压就是在指定维度索引上插入维度1