【PyTorch】多次元配列の次元（軸）を任意の順番に入れ替えるtorch.permute

PyTorchで多次元配列の次元（軸）を任意の順番に入れ替えるtorch.permuteを使う。

• torch.permute – PyTorch v1.11.0documentation

torch.permuteの使い方

第一引数inputに処理する配列Tesonr、第二引数dimsに次元の順番をタプルで指定する。

import torch

x = torch.randn(2, 3, 5, 7)
print(x.shape)
# torch.Size([2, 3, 5, 7])

x = torch.permute(x, (2, 0, 3, 1))
print(x.shape)
# torch.Size([5, 2, 7, 3])

入力の次元番号をタプルで指定することで、次元が入れ替わっていることが確認できる。

Numpyで同様のことをするには

Numpyで同様のことをするには、transposeを使うようです。

• numpy.transpose – Numpy API reference

関連記事、参考記事

• PyTorch 記事一覧

PyTorch公式からも紹介されている本で、基礎的な内容から実際の画像データを用いた実践的な内容まで網羅しています。

• PyTorch実践入門