Fonction torch.poisson de PyTorch

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

torch.poissonC'est une fonction de PyTorch utilisée pour générer des nombres aléatoires à partir d'une distribution de Poisson.

Définition de fonction

torch.poisson(lam, generator=None, out=None)

Description du paramètre

  • lam- Le paramètre lambda (valeur attendue/moyenne) de la distribution de Poisson doit être non négatif
  • generator- Générateur de nombres aléatoires (facultatif)
  • out- Tenseur de sortie (facultatif)

Exemple d'utilisation

Exemple

import torch

# Générer un tenseur aléatoire en utilisant une seule valeur lambda
result1 = torch.poisson(lam=5.0, size=(3, 3))
print("tenseur aléatoire 3x3 de loi de Poisson avec lambda=5 :")
print(result1)

# Utiliser un tenseur comme paramètre lambda
lam_tensor = torch.tensor([1.0, 2.0, 5.0, 10.0])
result2 = torch.poisson(lam_tensor)
print("nRésultat d'échantillonnage en utilisant le tenseur lambda :")
print(result2)

# Simuler des données de comptage
lam = torch.tensor([0.5, 1.0, 2.0, 5.0, 10.0])
samples = torch.poisson(lam)
print("nÉchantillonnage de Poisson avec différentes valeurs de lambda :")
for i, (l, s) in enumerate(zip(lam.tolist(), samples.tolist())):
    print(f" lambda={l}: {s}")

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

Autres extensions