2016-02-28 13 views
9

Tensorflowで次のものと同等のものは何ですか?Tensorflowで列合計を行う方法は?

np.sum(A, axis=1) 
+2

[reduce_sum](https://www.tensorflow.org/versions/r0.7/api_docs/python/math_ops.html#reduce_sum)のためにドキュメントをチェック –

答えて

24

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 
+0

私が探していたまさにです、ありがとう!なぜこれはウェブサイトのハウツーではありませんか? – maroxe

+0

'tf.reduce_sum(x、[0、1])==> 6'という理由を少し詳しくお聞かせください。私はそれによってうまくいった。 – lerner

+1

@Lemer - TFに2番目の軸(0番目と1番目の軸)の合計を求めるので、行列が2Dなのですべての要素の完全な合計が得られます。一般に、KDテンソルを持ち、L軸上で合計すると、(K-L)Dテンソルになります。したがって、K = Lの場合、常に浮動小数点(0Dテンソル)を出力します。 – lejlot

関連する問題