keras将2d张量转换为3d偏移的时间步长

时间:2018-11-14 19:38:58

标签: python keras backend lstm

我得到一个2d张量,我想将其转换为3D并将其添加到3d lstm输入张量(样本,时间步长,特征)。 我在2d到3d中使用expand K.expand_dims和K.repeat_elements。 问题是每次重复元素要匹配我的输入时间步时,我都需要从1步开始转换... 在这里,我的代码丢失了。

main_inputs = Input(shape=(30, 6))
lstm1_out = LSTM(100, return_sequences=False)(main_inputs)
output1= Dense(1)(lstm1_out)
output1_mod = K.expand_dims(output1,2) 
output1_mod = K.repeat_elements(output1_mod, 30, axis=1)
merge_input = K.concatenate([main_inputs, output1_mod],2)
lstm2_out = LSTM(100, return_sequences=False)(merge_input)
finaloutput= Dense(1)(lstm2_out)
...

我真的不知道该怎么做...

0 个答案:

没有答案