tf.reduce_sum()函数
tf.reduce_sum(input_tensor,axis=None,keepdims=None)
imput_tensor:输入tensor
axis:指定计算轴，可设定值为0，1，2，....
keepdimis:是否保持原有tensor的维度
reduce_sum通常用于tensor在指定维度方向的求和。例子如下:
现在有一个2 * 3 * 4的tensor:
[[[ 1,...
原创
2019-11-21 16:25:57 ·
333 阅读 ·
0 评论
关注
分享
扫一扫