如何在TensorFlow中交换张量轴？

Question

我有一个形状(30, 116, 10)的张量，我想交换前两个维度，所以我有一个形状的张量(116, 30, 10)

我看到numpy实现了这样一个函数（np.swapaxes），我在tensorflow中搜索了类似的东西，但我一无所获。

你有什么主意吗？

Answer 1

tf.transpose提供与np.swapaxes相同的功能，尽管它采用更通用的形式。在你的情况下，你可以做tf.transpose(orig_tensor, [1, 0, 2])，相当于np.swapaxes(orig_np_array, 0, 1)。