如何在TensorFlow图中命名除法运算？

Question

我在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")

但确切的语法不包括在内。

Answer 1

tf.div，tf.truediv和tf.floordiv都采用name参数。根据您想要的除法运算符，其中一个可能是您正在寻找的。