HingeEmbeddingLoss

class torch.nn.HingeEmbeddingLoss(margin=1.0, size_average=None, reduce=None, reduction='mean') [source]

Измеряет потерю, заданную тензором входа $x$ и тензором меток $y$ (содержащим 1 или -1). Обычно используется для измерения сходства или различия двух входов, например, используя L1 парное расстояние как $x$, и обычно используется для обучения нелинейных вложений или полуконтрольного обучения.

Функция потерь для $n$-й выборки в мини-пачке:

$$l_n = \begin{cases} x_n, & \text{if}\; y_n = 1,\\ \max \{0, margin - x_n\}, & \text{if}\; y_n = -1, \end{cases}$$

а общая функция потерь

$$\ell(x, y) = \begin{cases} \operatorname{mean}(L), & \text{if reduction} = \text{`mean';}\\ \operatorname{sum}(L), & \text{if reduction} = \text{`sum'.} \end{cases}$$

где $L = \{l_1,\dots,l_N\}^\top$.

Параметры

• margin (float, необязательно) – Имеет значение по умолчанию 1.
• size_average (bool, необязательно) – Устаревшее (см. reduction). По умолчанию потери усредняются по каждому элементу потерь в пачке. Обратите внимание, что для некоторых потерь есть несколько элементов на выборку. Если поле size_average установлено в False, потери вместо этого суммируются для каждой мини-пачки. Игнорируется, когда reduce равно False. Значение по умолчанию: True
• reduce (bool, необязательно) – Устаревшее (см. reduction). По умолчанию потери усредняются или суммируются по наблюдениям для каждой мини-пачки в зависимости от size_average. Когда reduce равно False, возвращает потерю на элемент пачки вместо этого и игнорирует size_average. Значение по умолчанию: True
• reduction (str, необязательно) – Указывает уменьшение, применяемое к выходу: 'none' | 'mean' | 'sum'. 'none': уменьшение не будет применено, 'mean': сумма выхода будет разделена на количество элементов в выходе, 'sum': выход будет суммирован. Примечание: size_average и reduce находятся в процессе устаревания, и в настоящее время указание любого из этих двух аргументов переопределит reduction. Значение по умолчанию: 'mean'

Форма:

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