PyTorch torch.bernoulli función

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

torch.bernoullies una función en PyTorch utilizada para generar números aleatorios a partir de la distribución de Bernoulli.

Definición de la función

torch.bernoulli(input, *, generator=None, out=None)
torch.bernoulli(p, size, *, generator=None, out=None)

Descripción de parámetros

  • input- Valor de probabilidad o tensor que contiene probabilidades (cada elemento representa la probabilidad de que la posición correspondiente sea 1)
  • p- Valor de probabilidad (se usa cuando input no es un tensor)
  • size- Forma del tensor de salida
  • generator- Generador de números aleatorios (opcional)
  • out- Tensor de salida (opcional)

Ejemplo de uso

Ejemplo

import torch

# Generar números aleatorios de distribución de Bernoulli usando un tensor de probabilidades
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Tensor de probabilidades:", probs)
print("Resultado del muestreo de Bernoulli:", result)

# Generar un tensor aleatorio con una probabilidad fija
result2 = torch.bernoulli(0.5, (3, 3))
print("Tensor aleatorio 3x3 generado con probabilidad fija 0.5:")
print(result2)

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

Otras extensiones