Функция torch.randperm в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.randpermявляется функцией в PyTorch, используемой для создания случайной перестановки целых чисел от 0 до n-1.

Определение функции

torch.randperm(n, dtype, device, requires_grad)

Пример использования

Пример

import torch

# Создать случайную перестановку от 0 до 9
x = torch.randperm(10)

print(x)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения