tf.reduce_sum(input_tensor,axis=None,keepdims=None)
reduce_sum通常用于tensor在指定维度方向的求和。例子如下:
现在有一个2 * 3 * 4的tensor:
[[[ 1, 2, 3, 4],
[ 5, 6, 7, 8],
[ 9,10,11,12]],
[[13,14,15,16],
[17,18,19,20],
[21,22,23,24]]]
当axis=0时,按第一个维度2求和,结果就应该是3 * 4的:
[[ 1+13, 2+14, 3+15, 4+16],
[ 5+17, 6+18, 7+19, 8+20],
[ 9+21,10+22,11+23,12+24]]
当axis=1时,按第一个维度2求和,结果就应该是2 * 4的:
[[ 1+ 5+ 9, 2+ 6+10, 3+ 7+11, 4+ 8+12],
[13+17+21,14+18+22,15+19+23,16+20+24]]
当axis=2时,按第一个维度2求和,结果就应该是2 * 3的:
[[ 1+ 2+ 3+ 4,
5+ 6+ 7+ 8,
9+10+11+12,
[13+14+15+16,
17+18+19+20,
21+22+23+24]]