交換張量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)。