Fonction torch.randint de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.randintC'est une fonction de PyTorch utilisée pour créer des tenseurs d'entiers aléatoires.

Définition de fonction

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

Exemples d'utilisation

Exemple

import torch

# Créer un tenseur d'entiers aléatoire de 0 à 9
x = torch.randint(0, 10, (3, 4))

print(x)

Le résultat de sortie est:

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

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions