美文网首页
math_ops.reduce_sum()函数

math_ops.reduce_sum()函数

作者: 是neinei啊 | 来源:发表于2017-12-08 15:47 被阅读118次

    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.

    相关文章

      网友评论

          本文标题:math_ops.reduce_sum()函数

          本文链接:https://www.haomeiwen.com/subject/ohieixtx.html