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'.

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