python - Tensorflow 合并维度和反向合并

Question

假设我有一个形状为 [3, 5, 200, 300, 3] 的张量。我怎样才能将前两个 dim 合并在一起，以便我得到形状为 [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))