PoissonNLLLoss

class torch.nn.PoissonNLLLoss(log_input=True, full=False, size_average=None, eps=1e-08, reduce=None, reduction='mean') [исходный код]

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

Функцию потерь можно описать следующим образом:

$$\text{target} \sim \mathrm{Poisson}(\text{input}) \text{loss}(\text{input}, \text{target}) = \text{input} - \text{target} * \log(\text{input}) + \log(\text{target!})$$

Последний член можно опустить или аппроксимировать с помощью формулы Стирлинга. Аппроксимация используется для значений target больше 1. Для значений target, меньших или равных 1, к функции потерь добавляются нули.

Параметры:

• log_input (bool, необязательно) – если True, функция потерь вычисляется как $\exp(\text{input}) - \text{target}*\text{input}$, если False, функция потерь равна $\text{input} - \text{target}*\log(\text{input}+\text{eps})$.

• full (bool, необязательно) –

  вычислять ли полную функцию потерь, то есть добавлять ли член с аппроксимацией Стирлинга

  $$\text{target}*\log(\text{target}) - \text{target} + 0.5 * \log(2\pi\text{target}).$$
• size_average (bool, необязательно) – Устарел (см. reduction). По умолчанию значения функции потерь усредняются по каждому элементу функции потерь в пакете. Обратите внимание, что для некоторых функций потерь на один образец приходится несколько элементов. Если для поля size_average задано значение False, вместо этого значения функции потерь суммируются для каждого мини-пакета. Игнорируется, если reduce равно False. По умолчанию: True
• eps (float, необязательно) – Небольшое значение, предотвращающее вычисление $\log(0)$, когда log_input = False. По умолчанию: 1e-8
• reduce (bool, необязательно) – Устарел (см. reduction). По умолчанию значения функции потерь усредняются или суммируются по наблюдениям для каждого мини-пакета в зависимости от size_average. Если reduce равно False, вместо этого возвращается значение функции потерь для каждого элемента пакета, а size_average игнорируется. По умолчанию: True
• reduction (str, необязательно) – Задает способ сведения выходных данных: 'none' | 'mean' | 'sum'. 'none': сведение не выполняется, 'mean': сумма выходных данных делится на количество их элементов, 'sum': выходные данные суммируются. Примечание: size_average и reduce постепенно выводятся из употребления; пока что указание любого из этих двух аргументов переопределяет reduction. По умолчанию: 'mean'

Примеры

>>> loss = nn.PoissonNLLLoss()
>>> log_input = torch.randn(5, 2, requires_grad=True)
>>> target = torch.randn(5, 2)
>>> output = loss(log_input, target)
>>> output.backward()

Форма:

• Входные данные: $(*)$, где $*$ означает любое количество измерений.
• Целевые данные: $(*)$, той же формы, что и входные данные.
• Выходные данные: по умолчанию скаляр. Если reduction равно 'none', то $(*)$, той же формы, что и входные данные.

forward(log_input, target) [исходный код]

Выполняет прямой проход.

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

Tensor