PyTorch torch.bernoulli Funktion
PyTorch torch-Referenzhandbuch
torch.bernoulliist eine Funktion in PyTorch, die verwendet wird, um Zufallszahlen aus der Bernoulli-Verteilung zu erzeugen.
Funktionsdefinition
torch.bernoulli(input, *, generator=None, out=None) torch.bernoulli(p, size, *, generator=None, out=None)
Parameterbeschreibung
input- Der Wahrscheinlichkeitswert oder ein Tensor, der Wahrscheinlichkeiten enthält (jedes Element stellt die Wahrscheinlichkeit dar, dass die entsprechende Position 1 ist)p- Der Wahrscheinlichkeitswert (wird verwendet, wenn die Eingabe kein Tensor ist)size- Die Form des Ausgabetensorsgenerator- Zufallszahlengenerator (optional)out- Ausgabetensor (optional)
Anwendungsbeispiel
Beispiel
import torch
# Bernoulli-verteilte Zufallszahlen mithilfe eines Wahrscheinlichkeits-Tensors generieren
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Wahrscheinlichkeits-Tensor:", probs)
print("Bernoulli-Stichprobenergebnis:", result)
# Einen Zufalls-Tensor mithilfe einer festen Wahrscheinlichkeit generieren
result2 = torch.bernoulli(0.5, (3, 3))
print("3x3-Zufallstensor, generiert mit fester Wahrscheinlichkeit 0.5:")
print(result2)
# Bernoulli-verteilte Zufallszahlen mithilfe eines Wahrscheinlichkeits-Tensors generieren
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Wahrscheinlichkeits-Tensor:", probs)
print("Bernoulli-Stichprobenergebnis:", result)
# Einen Zufalls-Tensor mithilfe einer festen Wahrscheinlichkeit generieren
result2 = torch.bernoulli(0.5, (3, 3))
print("3x3-Zufallstensor, generiert mit fester Wahrscheinlichkeit 0.5:")
print(result2)
Andere Erweiterungen