Tensorflow2.x中的Flatten()层,虽然名字叫展平,我们理解上以为是将其展平为一维张量,但是实际上它还是会保留batch_size这个维度。记录一个例子。 1.shape为 [2,3] 结果是不会展平 2. shape为[2,3,3] 结果是会展平为[2,9]