如何从Tensorflow模式获得张量值

时间:2017-10-08 18:24:42

标签: tensorflow deep-learning

我正在使用Seq2Seq模型的以下实现。现在,如果我想传递一些输入并获得编码器隐藏状态(self.encoder_last_state)的相应值,我该怎么办呢?

https://github.com/JayParks/tf-seq2seq/blob/master/seq2seq_model.py

1 个答案:

答案 0 :(得分:1)

您需要先组装input_feed,类似于预测例程。完成后,只需在所需的隐藏层上执行sess.run。

要解释input_feed:

input_feed = self.check_feeds(encoder_inputs, encoder_inputs_length, decoder_inputs=None, decoder_inputs_length=None, decode=True)
input_feed[self.keep_prob_placeholder.name] = 1.0

sess.run self.encoder_last_state

encoder_last_state_activations = sess.run(self.encoder_last_state, input_feed)