方法定义:
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,其他维度形状不变。
版权声明:
本文来源网络,所有图片文章版权属于原作者,如有侵权,联系删除。
本文网址:https://www.mushiming.com/mjsbk/510.html