reshape的参数使用

经常回会忘记reshape的参数是什么意思.

例如 

x.shape
Out[14]: torch.Size([1000])

y=x.reshape((-1, 1))

那么

y.Shape

Out[14]: torch.Size([1000,1])

第一个参数是行数,第二个参数是列数,如果设置成-1,就是根据另一个参数的和原本大小自动计算另一个参数化值.

也即是这里

y=x.reshape((1000, 1))

得到的是一样的结果.因为可以自动根据总大小1000,和另一个参数列数1来计算行数为1000.