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矩阵。