从模型获取Keras输入张量

时间:2018-10-12 19:40:37

标签: keras

我正在研究Keras示例中的序列到序列模型。

https://github.com/keras-team/keras/blob/master/examples/lstm_seq2seq.py

在此示例中,他们从输入张量构建模型。

encoder_inputs = Input(shape=(None, num_encoder_tokens))
model = Model([encoder_inputs, decoder_inputs], decoder_outputs)
print(encoder_inputs)

输出:

Tensor("input_1:0", shape=(?, ?, 71), dtype=float32)

一旦建立了模型,是否有一种方法可以从模型中检索输入张量?类似于

encoder_inputs = model.layers[0].??????

1 个答案:

答案 0 :(得分:0)

是的,这只是model.inputmodel.inputs