Función torch.randint de PyTorch

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

torch.randintEs una función de PyTorch utilizada para crear tensores aleatorios de enteros.

Definición de la función

torch.randint(low, high, size, dtype, device, requires_grad)

Ejemplo de uso

Ejemplo

import torch

# Crear un tensor de enteros aleatorios de 0 a 9
x = torch.randint(0, 10, (3, 4))

print(x)

El resultado de salida es:

tensor([[4, 5, 2, 6],
        [7, 9, 3, 1],
        [2, 8, 5, 4]])

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

Otras extensiones