pytorch每日一学39(torch.reshape())更改tensor为指定形状

第39个方法

torch.reshape(input, shape) → Tensor

此方法的作用是对input进行形状变化，形状变为shape，这个我们是会经常遇到的方法。
直接看用法：

方法非常的简单，直接将input和形状输入就可以了，但是一定要计算好维度，例如下面6个数字不能组成2x2的矩阵，所以报错。

为了避免这种情况，我们可以使用-1数字，使用-1的地方表示系统根据tensor中的元素和你前面给定的一些元素自动填写-1的地方，如下图：

总元素为6个，指定形状给了一个2，另一个就是系统自动根据6/2=3来算出。

但是注意，即使是使用-1也不是所有的都可用，例如3个元素无论如何都不能凑成2与一个数字相乘，所以还是要注意自己的shape矩阵。

注意，input中只能由一个数字为-1，不能出现两个或者多个-1的情况，因为那样的话不能根据其它的数字来进行算出-1那里具体的数字，如下所示：

• 当然，也可以直接使用Tensor.reshape来更改形状，这样的话就不用使用torch了：
  

当然，这样过后a不变，只是返回了一个新的变量