在上一篇博客中,我们简单分析了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",博主也不知。欢迎懂的朋友可以在评论区留言