tensorflow2.x的Flatten()

Tensorflow2.x中的Flatten()层，虽然名字叫展平，我们理解上以为是将其展平为一维张量，但是实际上它还是会保留batch_size这个维度。记录一个例子。

1.shape为 [2,3]

结果是不会展平

2. shape为[2,3,3]

结果是会展平为[2,9]