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

3
我正在使用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")

但是确切的语法我不太清楚。

我不熟悉TensorFlow,但是在谷歌搜索“tensorflow division”后发现tf.divtf.truedivtf.floordiv都需要一个name参数。这是你要找的吗? - user2357112
谢谢。我不知道我怎么会忽略那个。你应该将它作为一个答案提出来。 - W.P. McNeill
1个回答

2
tf.divtf.truedivtf.floordiv都需要一个name参数。根据你想要的除法运算符,其中之一可能是你要找的。请注意保留HTML标签。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接