码迷,mamicode.com
首页 > 其他好文 > 详细

tf.reduce_sum()

时间:2018-05-07 13:30:21      阅读:250      评论:0      收藏:0      [点我收藏+]

标签:维度   none   cti   inpu   col   div   keep   保留   als   

#axis 表示在哪个维度进行sum操作,不写代表所有维
#keep_dims 是否保留原始数据维度
reduce_sum(
    input_tensor,
    axis=None,
    keep_dims=False,
    name=None,
    reduction_indices=None
)

官方例子:

# ‘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

 

tf.reduce_sum()

标签:维度   none   cti   inpu   col   div   keep   保留   als   

原文地址:https://www.cnblogs.com/helloworld0604/p/9001782.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!