LogSoftmax

class torch.nn.LogSoftmax(dim=None) [source]

Применяет функцию $\log(\text{Softmax}(x))$ к тензору входных данных размерности n. Формула LogSoftmax может быть упрощена как:

$$\text{LogSoftmax}(x_{i}) = \log\left(\frac{\exp(x_i) }{ \sum_j \exp(x_j)} \right)$$

Форма:

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

Параметры

dim (int) – Измерение, по которому будет вычисляться LogSoftmax.

Возвращает

Тензор той же размерности и формы, что и входные данные, со значениями в диапазоне [-бесконечность, 0)

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

None

Примеры:

>>> m = nn.LogSoftmax(dim=1)
>>> input = torch.randn(2, 3)
>>> output = m(input)