在Tensorflow Keras functional API中展平整个批次

Question

我想在我的网络中伪造整个批次，以进一步处理数据。

这是一个我想要完全展平的层的例子(包括批次维度)

x = tf.keras.layers.Dense(128, activation='relu')(x) # has shape [None, 128]
x = Flatten(shape=(-1)) # Example function, should have shape [batch_size x 128]
# process x
x = Unflatten(shape=[None, 128]) # Example function, has again shape [None, 128]
# ...

我找到了this的帖子，似乎并没有涵盖Tensorflow 2.x。如何在TF 2.x中展平整个批次？

Answer 1

Tensorflow2提供了一个重整层。(https://www.tensorflow.org/api_docs/python/tf/keras/layers/Reshape)。