Función torch.randperm de PyTorch


Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.randpermEs una función en PyTorch utilizada para crear una permutación aleatoria de 0 a n-1.

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

# Crear una permutación aleatoria de 0 a 9
x = torch.randperm(10)

print(x)

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones