torch.nn.functional.gaussian_nll_loss

torch.nn.functional.gaussian_nll_loss(input, target, var, full=False, eps=1e-06, reduction='mean') [source]

Функция потерь отрицательного логарифма правдоподобия гауссовой модели.

Подробности см. в GaussianNLLLoss.

Параметры

input (Tensor) – математическое ожидание гауссова распределения.
target (Tensor) – выборка из гауссова распределения.
var (Tensor) – тензор положительных дисперсий, по одному для каждого ожидания в входном массиве (гетероскедастичный), или одна общая (гомоскедастичная).
full (bool, optional) – включать постоянный член в вычисление потерь. По умолчанию: False.
eps (float, optional) – значение, добавляемое к var для устойчивости. По умолчанию: 1e-6.
reduction (str, optional) – указывает способ обработки вывода: 'none' | 'mean' | 'sum'. 'none': без обработки, 'mean': среднее значение потерь по всем элементам батча, 'sum': сумма потерь по всем элементам батча. По умолчанию: 'mean'.

Тип возвращаемого значения