tf.reduce_sum(input_tensor, reduction_indices=None, keep_dims=False, name=None)
计算tensor命令某个维度的值的和
举个🌰栗子：
For example:
# 'x' is [[1, 1, 1]]
# [1, 1, 1]]
tf.reduce_sum(x) ==> 6
tf.reduce_sum(x, 0) ==> [2, 2, 2]
tf.reduce_sum(x, 1) ==> [3, 3]
tf.reduce_sum(x, 1, keep_dims=True) ==> [[3], [3]]
tf.reduce_sum(x, [0, 1]) ==> 6
参数:
input_tensor: 需要reduce的tensor.
reduction_indices: reduce的维度.如果缺省，默认all.
keep_dims: If true, retains reduced dimensions with length 1.
name: A name for the operation (optional).
返回值:
the reduced tensor.
网友评论