Keras - RepeatVector 層



RepeatVector 用於重複輸入 n 次。例如,如果將帶有引數 16RepeatVector 應用於具有輸入形狀為 (batch_size, 32) 的層,則該層的輸出形狀將為 (batch_size, 16, 32)

RepeatVector 有一個引數,如下所示 -

keras.layers.RepeatVector(n)

使用 RepeatVector 層的一個簡單示例如下 -

>>> from keras.models import Sequential 
>>> from keras.layers import Activation, Dense, RepeatVector 
>>> 
>>> 
>>> model = Sequential() 
>>> layer_1 = Dense(16, input_shape=(8,)) 
>>> model.add(layer_1) 
>>> layer_2 = RepeatVector(16) 
>>> model.add(layer_2) 
>>> layer_2.input_shape (None, 16) 
>>> layer_2.output_shape
(None, 16, 16)
>>>

其中將16 設定為重複次數。

廣告