MSELoss

class torch.nn.MSELoss(size_average=None, reduce=None, reduction='mean') [исходный код]

Создает критерий, который измеряет среднеквадратичную ошибку (квадрат нормы L2) между каждым элементом входных данных $x$ и целевым значением $y$.

Нередуцированная функция потерь (то есть при reduction, установленном в 'none') может быть описана следующим образом:

$$\ell(x, y) = L = \{l_1,\dots,l_N\}^\top, \quad l_n = \left( x_n - y_n \right)^2,$$

где $N$ — размер пакета. Если reduction не равно 'none' (значение по умолчанию — 'mean'), то:

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

$x$ и $y$ — тензоры произвольной формы, каждый из которых содержит в общей сложности $N$ элементов.

Операция усреднения по-прежнему выполняется по всем элементам с делением на $N$.

Деления на $N$ можно избежать, если задать reduction = 'sum'.

Параметры:

• 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'

Форма:

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

Примеры

>>> loss = nn.MSELoss()
>>> input = torch.randn(3, 5, requires_grad=True)
>>> target = torch.randn(3, 5)
>>> output = loss(input, target)
>>> output.backward()

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

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

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

Tensor