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, 2, 3, 4],    
  [ 5, 6, 7, 8],      
  [ 9,10,11,12]],    
 [[13,14,15,16],
  [17,18,19,20],
  [21,22,23,24]]]   

当axis=0时,按第一个维度2求和,结果就应该是3 * 4的:

[[ 1+13, 2+14, 3+15, 4+16],    
 [ 5+17, 6+18, 7+19, 8+20],      
 [ 9+21,10+22,11+23,12+24]]   

当axis=1时,按第一个维度2求和,结果就应该是2 * 4的:

[[ 1+ 5+ 9, 2+ 6+10, 3+ 7+11, 4+ 8+12],    
 [13+17+21,14+18+22,15+19+23,16+20+24]]   

当axis=2时,按第一个维度2求和,结果就应该是2 * 3的:

[[ 1+ 2+ 3+ 4,    
   5+ 6+ 7+ 8,      
   9+10+11+12,    
 [13+14+15+16,
  17+18+19+20,
  21+22+23+24]]   

 

你可能感兴趣的:(tensorflow)