将名称分配给 TensorFlow 图中的 RNN 下一个状态答案

【问题标题】：Assign name to an RNN next state in a TensorFlow graph将名称分配给 TensorFlow 图中的 RNN 下一个状态
【发布时间】：2016-10-08 18:33:15
【问题描述】：

我有一个 TensorFlow RNN 模型，我在其中进行以下调用

outputs, next_state = tf.nn.dynamic_rnn(rnn_layers, embedded_input, initial_state=state)

我想为next_state 变量命名，以便以后可以使用get_tensor_by_name 检索它。有没有简单的方法可以做到这一点？

【问题讨论】：

【解决方案1】：

我觉得它应该已经有了一个名字（如果你还没有的话，你应该验证一下）。

如果没有，以下是如何有效地为张量分配新名称： https://stackoverflow.com/a/34399966/447599

【讨论】：

【解决方案2】：

重命名操作的典型方法是使用tf.identity(),，它几乎没有运行时开销：

outputs, next_state = tf.nn.dynamic_rnn(rnn_layers, embedded_input, initial_state=state)
next_state = tf.identity(next_state, name='next_state_node')

稍后您可以通过以下方式获取next_state：

lstm_state = sess.graph.get_tensor_by_name('next_state_node:0')

【讨论】：