深度学习

Keras之layer: RepeatVector和TimeDi

2021-07-19  本文已影响0人  Qiangest

Layer: RepeatVector

RepeatVector(10): 把向量复制10次. 比如原先的向量是(32, ) 之后就变成了(32,10)

主要用于LSTM encoder. encoder得到的结果是一个一维feature向量, 但是decoder回去也是LSTM, 所以必须先重复10次当作是有时间步的, 才能输入decoder的LSTM中去.

Layer: TimeDistributed

TimeDistributed( Dense( 12 , input_shape=( 10 , 6 ) )) 把dense layer应用到输入的每个时间步里去. 比如之前是(13,10,6), 这么操作之后就变成了(13,10,12).

TimeDistributed不会改变维度, 但是会改变feature的个数.

上一篇 下一篇

猜你喜欢

热点阅读