tensorflow tf.reduce_mean 的理解
2017-12-26 本文已影响0人
kernqooy
本来很简单的一个概念花了很长时间理解,用了程序上跑了很多次越跑越看不懂。网上的文章解释不也是很好。最后突然演算出来了,以下是我的理解。
第一个 tensor shape : 2 x 3
0 和 1 是index ,分别对应的是2 和 3
所以 index为0时 reduce 过后是 0 x 3
index 为 1 时 reduce 过后是 2 x 0 注意不是 2 x 2 也不是 2 x 1