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 设置为重复次数。

广告
© . All rights reserved.