reshape中的-1,x = x.reshape(-1,x.size(-1)*1)

arr.shape    # (a,b)
arr.reshape(m,-1) #改变维度为m行、d列 (-1表示列数自动计算,d= a*b /m )
arr.reshape(-1,m) #改变维度为d行、m列 (-1表示行数自动计算,d= a*b /m )

16个数,行数为4,列数自动计算。

import numpy as np
arr = np.arange(16).reshape(4, -1)
print(arr)

reshape中的-1,x = x.reshape(-1,x.size(-1)*1)_第1张图片

你可能感兴趣的:(Pytorch,Python,numpy,python)