tf.transposeを使わないTensorFlowでのテンソル軸の入れ替え

Question

テンソルは未知の形状ですが、少なくとも3次元です。すなわち、shape=[a, b, c, ...]です。 私はこれは動作しますが、醜いです（this questionで提案されているように、私は、tf.transposeを使用することはできません）テンソルがどのくらい知らなくても、寸法aとbを切り替える

Answer 1

をしたいと思います：

tf.transpose(x, [1, 0] + [i+2 for i in range(tf.shape(x).shape[0]-2)])