Tensorflow合并尺寸和反向合并

Question

假设我有一个张量形状[3, 5, 200, 300, 3]。我怎样才能合并前两个暗淡在一起，以便我有张量的形状[15, 200, 300, 3]。然后，我可以反向合并操作并具有原始形状。

Answer 1

你可以使用tf.reshape

a = tf.random_normal(shape=(3, 5, 200, 300, 3))
b = tf.reshape(a, shape=(15, 200, 300, 3))
...
c = tf.reshape(b, shape=(3, 5, 200, 300, 3))