用于形状数组的负指数是多少？(Tensorflow)

Question

在Tensorflow的MNIST教程中,我们将最后一个Pool层的输出重新整形为单个向量.编写的代码是:

h_pool2_flat = tf.reshape(h_pool2, [-1, 7*7*64]) 

什么是-1指数？我们不是只想将输出整形为单个向量,那么为什么我们不仅仅重塑为:[1,7*7*64]？

提前致谢!

Answer 1

-1表示自动扩展.例如,具有[-1,7*7*64]的重塑将将[19*7*7*64]的1维形状转换为[19,7*7]的二维形状.*64].

另一个例子,具有[5,-1,7]的重塑将将例如[70]的1维形状转换为[5,2,7]的3维形状.

Answer 2

-1 将被推断。

因此 NumPy 或 TensorFlow 将根据传入的元素和指定的其他维度推断维度大小应该是多少