Skip to content

Reshape layer issue

Created by: Mmiglio

Hi all,

I'm having trouble while compiling a model similar to this

model = tf.keras.models.Sequential([
        tf.keras.layers.Input((10)),
        tf.keras.layers.Dense(10*3),
        tf.keras.layers.Reshape((10,3)),
        tf.keras.layers.Softmax()
])

These are the errors:

Screenshot 2020-12-14 at 11 59 57

Screenshot 2020-12-14 at 12 00 28

My guess is that the problem is someway related to the combination of Reshape + softmax. Thanks in advance for the help!