如何在Tensorflow中創建列總和？

Tensorflow中的以下等效內容是什麼？如何在Tensorflow中創建列總和？

np.sum(A, axis=1)

來源

2016-02-28 maroxe

檢查文檔爲[reduce_sum（https://www.tensorflow.org/versions/r0.7/api_docs/python/math_ops.html#reduce_sum） –

有tf.reduce_sum這是一個更強大的工具這樣做。 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

來源

2016-02-28 23:42:13 lejlot

這正是我一直在尋找，謝謝！爲什麼這不是在網站howto？ – maroxe

請詳細介紹一下爲什麼'tf.reduce_sum（x，[0，1]）==> 6'？我絆倒了它。 – lerner

@Lemer - 你要求TF總結兩個軸 - 第0和第1，所以既然矩陣是2D，你最終會得到所有元素的全部總和。一般情況下，KD張量和在L軸上求和的結果是（K-L）D張量，因此對於K = L，它總是輸出一個浮點數（0D張量）。 – lejlot

如何在Tensorflow中創建列總和？

回答

相關問題