【PyTorch】ランダムな整数の順列を生成するtorch.randperm【乱数】

Python

2022/07/192024/02/14

torch.randpermの使い方

ランダムな整数の順列を生成するには、torch.randperm()を使う。

引数にnを指定すると、0からn-1までの順列が作成され、ランダムに並び替えられる。

import torch

print(torch.randperm(4))
# tensor([1, 3, 2, 0])

print(torch.randperm(5))
# tensor([1, 0, 3, 2, 4])