Tensorflow解决reduce_mean出现"nan"提示

在上一篇博客中，我们简单分析了tensorflow参数传递的问题。接下来我们来看reduce_sum()

tf.reduce_sum(
    input_tensor, 
    axis=None, 
    keepdims=None,
    name=None,
    reduction_indices=None, 
    keep_dims=None)

reduce_sum中的reduction_indices都是针对某个轴算总和的意思，如果不设值，则是计算的全体元素之和。

这里就是我曾经犯下的问题所在，在之前的代码中，我没有设置reduction_indices=0，因此在计算reduce_mean的时候，算得值只是reduce_sum的值。

        self.loss = reduce_mean(-reduce_sum(mul(self.ys,log(self.prediction)),reduction_indices=[0]))

至于为什么是"nan"，博主也不知。欢迎懂的朋友可以在评论区留言