Функция torch.bernoulli PyTorch
Справочное руководство PyTorch torch
torch.bernoulli— это функция в PyTorch, используемая для генерации случайных чисел из распределения Бернулли.
Определение функции
torch.bernoulli(input, *, generator=None, out=None) torch.bernoulli(p, size, *, generator=None, out=None)
Описание параметра
input- Значение вероятности или тензор, содержащий вероятности (каждый элемент представляет вероятность того, что соответствующая позиция равна 1)p- Значение вероятности (используется, когда входные данные не являются тензором)size- Форма выходного тензораgenerator- Генератор случайных чисел (необязательно)out- Выходной тензор (необязательно)
Пример использования
Пример
import torch
# Генерация случайных чисел распределения Бернулли с использованием тензора вероятностей
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Тензор вероятностей:", probs)
print("Результат сэмплирования Бернулли:", result)
# Генерация случайного тензора с использованием фиксированной вероятности
result2 = torch.bernoulli(0.5, (3, 3))
print("3x3 случайный тензор, сгенерированный с фиксированной вероятностью 0.5:")
print(result2)
# Генерация случайных чисел распределения Бернулли с использованием тензора вероятностей
probs = torch.tensor([0.1, 0.5, 0.9])
result = torch.bernoulli(probs)
print("Тензор вероятностей:", probs)
print("Результат сэмплирования Бернулли:", result)
# Генерация случайного тензора с использованием фиксированной вероятности
result2 = torch.bernoulli(0.5, (3, 3))
print("3x3 случайный тензор, сгенерированный с фиксированной вероятностью 0.5:")
print(result2)
Другие расширения