如何重组张量

Question

在不失一般性的前提下，我具有以下3d张量

形状（2、3、3）

[[[1 2 3]
  [4 5 6]
  [7 8 9]]
 [[-1 -2 -3]
  [-4 -5 -6]
  [-7 -8 -9]]

我需要重新组织上述张量

[[[1 2 3]
  [-1 -2 -3]]
 [[4 5 6]
  [-4 -5 -6]]
 [[7 8 9]
  [-7 -8 -9]]]

即形状的张量（3 x 2 x 3）。我应该如何在tensorflow中做到这一点？

Answer 1

tf.transpose正是您想要的。

如果要将形状为[3,4,5]的a转置为[4,5,3]，则可以使用tf.transpose(a, [1, 2, 0])