张量流量：张量的反向平坦化

Question

OS：Kubuntu 14.04，Tensorflow版本= 0.12，Python版本= 2.7

在我的最后一层之后，我的神经网络的输出形状为[batch, height, width, 2]，在我的情况下为[64, 32, 256, 2]。

之后我使用

output = tf.contrib.layers.flatten(input=output)

生成形状为[64, 16384]

的Tensor

出于评估目的，我想在不同的函数中反转这种扁平化以获得原始的[64, 32, 256, 2] Tensor。

如何做到这一点？

Answer 1

只需reshape到你想要的形状：tf.reshape(output, [-1, 32, 256, 2])。