MultiLabelSoftMarginLoss

class torch.nn.modules.loss.MultiLabelSoftMarginLoss(weight=None, size_average=None, reduce=None, reduction='mean') [source]

Создает функцию потерь, оптимизирующую многометочную функцию потерь по принципу «один против всех» на основе максимальной энтропии между входом $x$ и целевым значением $y$ размера $(N, C)$. Для каждого образца в мини-пакете:

$$loss(x, y) = - \frac{1}{C} * \sum_i y[i] * \log((1 + \exp(-x[i]))^{-1}) + (1-y[i]) * \log\left(\frac{\exp(-x[i])}{(1 + \exp(-x[i]))}\right)$$

где $i \in \left\{0, \; \cdots , \; \text{x.nElement}() - 1\right\}$, $y[i] \in \left\{0, \; 1\right\}$.

Параметры:

• weight (Tensor, необязательно) – вес ручного масштабирования, назначаемый каждому классу. Если указан, это должен быть тензор размера C. В противном случае считается, что все его элементы равны единице.
• 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'

Форма:

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

Примеры

>>> loss = nn.MultiLabelSoftMarginLoss()
>>> input = torch.randn(3, 5, requires_grad=True)
>>> target = torch.empty(3, 5).random_(2)
>>> output = loss(input, target)
>>> output.backward()

forward(input, target) [source]

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

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

Tensor