标签： tensorflow deep-learning
我有一个未知形状的张量，但它至少是3维，即shape=[a, b, c, ...]。 我想切换尺寸a和b，而不知道张量有多长（所以我不能使用tf.transpose，如this question中所述）
shape=[a, b, c, ...]
a
b
答案 0 :(得分：2)
这很有效，但很难看：
tf.transpose(x, [1, 0] + [i+2 for i in range(tf.shape(x).shape[0]-2)])