我正在使用TensorFlow编写语言模型,按照
但是确切的语法我不太清楚。
ptb_word_lm.py
的示例进行。我计算批处理成本的方式如下: loss = tf.nn.seq2seq.sequence_loss_by_example([logits],
[tf.reshape(y, [-1])],
[tf.ones([batch_size * time_steps])],
name="loss")
cost = tf.reduce_sum(loss) / batch_size
我希望在图中给成本节点附加一个名称,以便我可以在TensorBoard中可视化并进行标量汇总。然而,我无法弄清如何将名称附加到/
中缀运算符上。我想我必须做一些类似于以下的操作:
cost = tf.DivisionOperator(cost, batch_size, name="cost")
但是确切的语法我不太清楚。
tf.div
,tf.truediv
和tf.floordiv
都需要一个name
参数。这是你要找的吗? - user2357112