[Python/Pytorch] tensor 랜덤으로 섞기

1. 랜덤 인덱스 순열 만들기

a = torch.tensor([1,2,3,4])
idx = 4 # 섞을 torch 길이
rand_idx = torch.randperm(idx)

torch.randperm(n) : 0 ~ (n-1) 사이의 값이 랜덤하게 들어있는 tensor를 출력한다.

2. 생성한 리스트의 인덱스에 맞게 원래의 tensor 섞기

shuffle_tensor = a[rand_idx]

결과