Fonction torch.bernoulli de PyTorch
Manuel de référence PyTorch torch
torch.bernoulliest une fonction dans PyTorch utilisée pour générer des nombres aléatoires à partir de la distribution de Bernoulli.
Définition de fonction
torch.bernoulli(input, *, generator=None, out=None) torch.bernoulli(p, size, *, generator=None, out=None)
Description du paramètre
input- La valeur de probabilité ou un tenseur contenant des probabilités (chaque élément représente la probabilité que la position correspondante soit 1)p- La valeur de probabilité (utilisée lorsque l'entrée n'est pas un tenseur)size- La forme du tenseur de sortiegenerator- Générateur de nombres aléatoires (facultatif)out- Tenseur de sortie (facultatif)
Exemple d'utilisation
Exemple
import torch
# Générer des nombres aléatoires de loi de Bernoulli à l'aide d'un tenseur de probabilités
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Tenseur de probabilités :", probs)
print("Résultat de l'échantillonnage de Bernoulli :", result)
# Générer un tenseur aléatoire à l'aide d'une probabilité fixe
result2 = torch.bernoulli(0.5, (3, 3))
print("tenseur aléatoire 3x3 généré avec une probabilité fixe de 0.5 :")
print(result2)
# Générer des nombres aléatoires de loi de Bernoulli à l'aide d'un tenseur de probabilités
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Tenseur de probabilités :", probs)
print("Résultat de l'échantillonnage de Bernoulli :", result)
# Générer un tenseur aléatoire à l'aide d'une probabilité fixe
result2 = torch.bernoulli(0.5, (3, 3))
print("tenseur aléatoire 3x3 généré avec une probabilité fixe de 0.5 :")
print(result2)
Autres extensions