如何在张量流中将张量从 `(None, a*b, None, ...)` 重塑为 `(None, a, b, None, ...)`？

Question

假设我有一个名为 t 的张量，其形状为 (None, a*b, None, ...)（...我的意思是它有更多的维度，但我不在乎）并且我想将其重塑为 {{1 }}。我该怎么办？

我认为我应该通过 (None, a, b, None, ...) 获得 t 的形状，然后对其进行处理，然后我应该使用 tf.shape(t) 来获得我想要的结果。但中间我该怎么办？