Функция torch.normal в PyTorch
Справочное руководство Pytorch torch
torch.normal— это функция в PyTorch, используемая для генерации случайных чисел из нормального распределения.
Определение функции
torch.normal(mean, std, *, generator=None, out=None) torch.normal(mean, std, size, *, generator=None, out=None)
Описание параметра
mean- Среднее значение нормального распределенияstd- Стандартное отклонение нормального распределенияsize- Форма выходного тензора (необязательно)generator- Генератор случайных чисел (необязательно)out- Выходной тензор (необязательно)
Пример использования
Пример
import torch
# Сгенерировать случайный тензор с фиксированным средним и стандартным отклонением
result1 = torch.normal(mean=0.0, std=1.0, size=(3, 3))
print("Среднее 0, стандартное отклонение 1 случайный тензор нормального распределения 3x3:")
print(result1)
# Использовать тензоры в качестве среднего и стандартного отклонения
mean = torch.tensor([0.0, 1.0, 2.0])
std = torch.tensor([1.0, 2.0, 3.0])
result2 = torch.normal(mean, std)
print("nРезультат сэмплирования с использованием тензорных параметров:")
print(result2)
# Использовать скалярное среднее и тензорное стандартное отклонение
result3 = torch.normal(mean=0.0, std=torch.tensor([1.0, 2.0, 3.0]))
print("nРезультат сэмплирования со скалярным средним и тензорным стандартным отклонением:")
print(result3)
# Сгенерировать случайный тензор с фиксированным средним и стандартным отклонением
result1 = torch.normal(mean=0.0, std=1.0, size=(3, 3))
print("Среднее 0, стандартное отклонение 1 случайный тензор нормального распределения 3x3:")
print(result1)
# Использовать тензоры в качестве среднего и стандартного отклонения
mean = torch.tensor([0.0, 1.0, 2.0])
std = torch.tensor([1.0, 2.0, 3.0])
result2 = torch.normal(mean, std)
print("nРезультат сэмплирования с использованием тензорных параметров:")
print(result2)
# Использовать скалярное среднее и тензорное стандартное отклонение
result3 = torch.normal(mean=0.0, std=torch.tensor([1.0, 2.0, 3.0]))
print("nРезультат сэмплирования со скалярным средним и тензорным стандартным отклонением:")
print(result3)
Другие расширения