在TensorFlow中交换张量轴而不使用tf.transpose

Question

我有一个未知形状的张量，但它至少是三维的，即shape=[a, b, c, ...]。 我想切换尺寸a和b，不知道张量有多长（所以我不能使用tf.transpose，如this question建议）

Answer 1

这工作，而是丑：

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