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)