本文共 396 字,大约阅读时间需要 1 分钟。
方法定义:
tf.reduce_sum( input_tensor, axis=None, keepdims=None, name=None)
tf.reduce_sum()作用是按一定方式计算张量中元素之和
input_tensor为待处理张量; axis指定按哪个维度进行加和,默认将所有元素进行加和; keepdims默认为False,表示不维持原来张量的维度,反之维持原张量维度; name用于定义该操作名字。 示例: 有如下张量,形状为(2,3,4)的三维张量: 分别按照axis=0,1,2进行加和的结果如下: 可以看到,原张量是三维的,指定axis进行一次tf.reduce_sum()后变成了二维张量。如果设置keepdims=True,则结果如下,可以看到计算完加和后的张量依然是三维的,其中指定axis的那个维度元素个数为1,其他维度形状不变。转载地址:http://wxrti.baihongyu.com/