ValueError: Input 0 of layer dense is incompatible with the layer: expected axis -1 of input shape to have value 784 but received input with shape (128, 28, 28)
源代码:
x = next(iter(test_db))
x_new_logits = model(x)
x_new = tf.sigmoid(x_rec_logits)
x_new = tf.reshape(x_new