Làm thế nào để thực hiện tổng cột trong Tensorflow?

Question

Tương đương với những điều sau đây trong Tensorflow là gì?

np.sum(A, axis=1) 

Answer 1

Có tf.reduce_sum là công cụ mạnh hơn một chút để làm như vậy. https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/api_docs/python/math_ops.md#tfreduce_suminput_tensor-reduction_indicesnone-keep_dimsfalse-namenone-reduce_sum

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