如何創建具有二維形狀的輸出密集層的 Keras 模型？

Question

我的模型的輸出是 (None, 2)，我需要它是 (None, 2, 14)。 使用以下模型，我應該更改什么以獲得所需的輸出形狀？

model = Sequential()
model.add(Dense(32, activation='relu', input_shape=(211,)))
model.add(Dropout(0.25))
model.add(Dense(32, activation='relu'))
model.add(Dense(2,activation='softmax'))

Answer 1

相反，最后一層的兩個神經元使用 28 個神經元，然后使用tf.keras.layer.Reshape()將(28,) to (2, 14)並得到你想要的：

import tensorflow as tf
model = tf.keras.Sequential()
model.add(tf.keras.layers.Dense(32, activation='relu', input_shape=(211,)))
model.add(tf.keras.layers.Dropout(0.25))
model.add(tf.keras.layers.Dense(32, activation='relu'))
model.add(tf.keras.layers.Dense(28,activation='softmax'))
model.add(tf.keras.layers.Reshape((2,14)))
model.summary()

輸出：

Model: "sequential"
_________________________________________________________________
 Layer (type)                Output Shape              Param #   
=================================================================
 dense (Dense)               (None, 32)                6784      
                                                                 
 dropout (Dropout)           (None, 32)                0         
                                                                 
 dense_1 (Dense)             (None, 32)                1056      
                                                                 
 dense_2 (Dense)             (None, 28)                924       
                                                                 
 reshape (Reshape)           (None, 2, 14)             0         
                                                                 
=================================================================
Total params: 8,764
Trainable params: 8,764
Non-trainable params: 0
_________________________________________________________________