通過池減少張量的維數[Tensorflow]

Question

我想用Tensorflow構建神經網絡體系結構。

我有一個變量，類型的張量。

說，

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

16是批量大小和一個輸入被編碼成20 尺寸，但有不同的數目的輸入。

在這裏，我怎樣才能改變它的維度到（16,20）只是意味着池的關於第二維有不同的大小。

謝謝。

Answer 1

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)