TensorFlowでテンソルの軸を交換するにはどうすればよいですか？

Question

形状のテンソル(30, 116, 10)、最初の2つの次元を交換したいので、形状のテンソル(116, 30, 10)

私はそのような関数が実装されているようにnumpyを見ました（np.swapaxes）テンソルフローに似たものを検索しましたが、何も見つかりませんでした。

何かアイデアはありますか？

Answer 1

_tf.transpose_ は、_np.swapaxes_と同じ機能を提供しますが、より一般的な形式です。あなたの場合、tf.transpose(orig_tensor, [1, 0, 2])を行うことができます。これはnp.swapaxes(orig_np_array, 0, 1)と同等です。