交换张量tensorflow

Question

轴我有形状（30,116,10）的张量，我想交换前两个dimesions，让我有形状的张量（116,30,10）

我看到作为这样一个功能（np.swapaxes）numpy和我在tensorflow搜索类似的东西，但我什么也没有找到。

你有什么想法吗？

由于提前，

亚历

Answer 1

tf.transpose提供相同的功能np.swapaxes，虽然在一个更一般化的形式。在你的情况下，你可以做tf.transpose(orig_tensor, [1, 0, 2])这将相当于np.swapaxes(orig_np_array, 0, 1)。