math_ops.reduce_sum()函数

2017-12-08  本文已影响118人  是neinei啊

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.

上一篇下一篇

猜你喜欢

热点阅读