Python Numpy中reshape函数参数-1的含义

z = np.array([[1, 2, 3, 4],[5, 6, 7, 8],[9, 10, 11, 12], [13, 14, 15,16]])
print(z.reshape(-1,1))

我们不知道z的shape属性是多少,但是想让z变成只有一列,行数不知道多少,通过z.reshape(-1,1),Numpy自动计算出有16行,新的数组shape属性为(16, 1),与原来的(4, 4)配套,即4x4=16个元素。
打印结果:
[[ 1]
[ 2]
[ 3]
[ 4]
[ 5]
[ 6]
[ 7]
[ 8]
[ 9]
[10]
[11]
[12]
[13]
[14]
[15]
[16]]

print(z.reshape(2,-1))

打印结果:
[[ 1 2 3 4 5 6 7 8]
[ 9 10 11 12 13 14 15 16]]

再如:

trainX = trainX.reshape([-1, 28, 28, 1])

我们不知道trainX的shape到底是什么,第一个参数未知,但是后面的形状参数知道,即[28,28,1],可以这么理解[28, 28, 1]是一个东西(我这里的实际情况中为图片,28行,28列,单通道),但是有多少个这样的东西,我们不知道,故把第一个参数设为-1,然后第一个可以根据真实的形状自适应得出当前情况下第一个参数的值。

参考:
https://blog.csdn.net/weixin_39449570/article/details/78619196
https://blog.csdn.net/cainiaohudi/article/details/79980826

你可能感兴趣的:(Python)