Función torch.poisson de PyTorch

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

torch.poissonEs una función de PyTorch utilizada para generar números aleatorios a partir de la distribución de Poisson.

Definición de la función

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

Descripción de parámetros

  • lam- El parámetro lambda de la distribución de Poisson (esperanza/media), debe ser un número no negativo
  • generator- Generador de números aleatorios (opcional)
  • out- Tensor de salida (opcional)

Ejemplo de uso

Ejemplo

import torch

# Generar un tensor aleatorio usando un único valor lambda
result1 = torch.poisson(lam=5.0, size=(3, 3))
print("Tensor aleatorio de distribución de Poisson 3x3 con lambda=5:")
print(result1)

# Usar un tensor como parámetro lambda
lam_tensor = torch.tensor([1.0, 2.0, 5.0, 10.0])
result2 = torch.poisson(lam_tensor)
print("nResultado del muestreo usando lambda como tensor:")
print(result2)

# Simular datos de conteo
lam = torch.tensor([0.5, 1.0, 2.0, 5.0, 10.0])
samples = torch.poisson(lam)
print("nMuestreo de Poisson con diferentes valores de lambda:")
for i, (l, s) in enumerate(zip(lam.tolist(), samples.tolist())):
    print(f"  lambda={l}: {s}")

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

Otras extensiones