Tensorflow2.0 函数式网络 Reshape layers

遇到一个reshpe的问题，记录一下

官方文档找到的一种解决方案

model = tf.keras.Sequential()
model.add(layers.Dense(7*7*256, use_bias=False, input_shape=(100,)))
model.add(layers.BatchNormalization())
model.add(layers.LeakyReLU())
model.add(layers.Reshape((7, 7, 256)))
assert model.output_shape == (None, 7, 7, 256)

Reference：https://www.tensorflow.org/tutorials/generative/dcgan

但我是用函数式写的model，上面那种写法不行
要用这种:

tf.reshape(one_layer, [-1, 256, num_class + 1])