通过合并[Tensorflow]减少Tensor的维数

Question

我正在尝试使用Tensorflow构建神经网络架构。

我有一个变量，类型为Tensor。

说，

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)