2017-05-14 21 views
0

私はTensorflowでニューラルネットワークアーキテクチャを構築しようとしています。プールでテンソルを縮小する[Tensorflow]

と私は可変タイプのTensorを持っています。

セイ、

a = <tf.Tensor shape(16, ?, 20) dtype=float32> 

16は、入力の異なる数が存在し、バッチサイズであり、入力が20 の寸法に符号化されるが。

ここでは、サイズが異なる2番目のディメンションを基準にして、単に平均的なプールによってディメンションを(16,20)に変更できます。

ありがとうございます。

答えて

2

reduce_mean?

a = tf.placeholder('float32', shape=(16, None, 20)) 
b = tf.reduce_mean(a, axis=1) 
print b 

出力:

Tensor("Mean:0", shape=(16, 20), dtype=float32) 
+0

まさに、私が欲しいもの。ありがとうございました –

関連する問題