Spec-Zone.ru › PyTorch 1

torch.nn.functional.poisson_nll_loss

torch.nn.functional.poisson_nll_loss(input, target, log_input=True, full=False, size_average=None, eps=1e-08, reduce=None, reduction='mean') [source]

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

См. PoissonNLLLoss для подробностей.

Параметры:
  • input (Тензор) – ожидаемое значение базового распределения Пуассона.
  • target (Тензор) – случайная выборка target∼Poisson(input)target \sim \text{Poisson}(input).
  • log_input (bool) – если True функция потерь вычисляется как exp⁡(input)−target∗input\exp(\text{input}) - \text{target} * \text{input}, если False тогда функция потерь равна input−target∗log⁡(input+eps)\text{input} - \text{target} * \log(\text{input}+\text{eps}). Значение по умолчанию: True
  • full (bool) – вычислять полную функцию потерь, т. е. добавлять член приближения Стирлинга. Значение по умолчанию: False target∗log⁡(target)−target+0.5∗log⁡(2∗π∗target)\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(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'
Тип возвращаемого значения:

Тензор

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.functional.poisson_nll_loss.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API