重塑数组时参数 (..., -1) 是什么意思？

Question

我遇到了以下声明：

train_set_x.reshape(train_set_x.shape[0], -1).T

train_set_x 的shape 是：(209, 64, 64, 3)

因此我认为shape[0] 是209，而T 是转置？

但是我无法理解上面的reshape 声明？什么是-1？

非常感谢任何澄清。

谢谢。

Answer 1

-1 将采用剩余的维度并将它们展平为 1 维。所以对于形状为(209, 64, 64, 3) 的数组调用：

arr.reshape(209, -1)

将产生一个形状为 (209, 12288) 或 (209, 64 * 64 * 3) 的矩阵

 >>> a = np.zeros([209, 64, 64, 3])
 >>> a.reshape(209, -1).shape
 (209, 12288)

如果您的代码用于 64 x 64 RGB 图像，您最终会将每个图像重新整形为一个长向量。

另外，请注意只有一个 -1 在数组将被重新整形为的新形状中。