Spec-Zone.ru › PyTorch 2

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/2.1/generated/torch.nn.functional.poisson_nll_loss.html

Spec-Zone.ru

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