【发布时间】:2018-11-21 17:51:53
【问题描述】:
我正在尝试使用 Keras 实现一个 seq2seq 编码器-解码器,编码器上的双向 lstm 如下:
from keras.layers import LSTM,Bidirectional,Input,Concatenate
from keras.models import Model
n_units = 8
n_input = 1
n_output = 1
# encoder
encoder_inputs = Input(shape=(None, n_input))
encoder = Bidirectional(LSTM(n_units, return_state=True))
encoder_outputs, forward_h, forward_c, backward_h, backward_c = encoder(encoder_inputs)
state_h = Concatenate()([forward_h, backward_h])
state_c = Concatenate()([forward_c, backward_c])
encoder_states = [state_h, state_c]
# decoder
decoder_inputs = Input(shape=(None, n_output))
decoder_lstm = LSTM(n_units*2, return_sequences=True, return_state=True)
decoder_outputs, _, _ = decoder_lstm(decoder_inputs, initial_state=encoder_states)
这是我在最后一行遇到的以下错误:
ValueError: Dimensions must be equal, but are 8 and 16 for
'lstm_2_1/MatMul_4' (op: 'MatMul') with input shapes: [?,8], [16,16].
有什么想法吗?
【问题讨论】:
-
您的代码在我的机器上运行没有任何错误(尽管在创建
Model对象并调用compile和fit之后)。你能发布你正在使用的完整代码吗?可能是你没有发的部分有问题。 -
你是对的,错误指向了这个块的最后一行,但是真正的错误是从与推理解码器相关的另一行传播的!谢谢,现在解决了!
标签: neural-network keras lstm rnn seq2seq