LogSoftmax

class torch.nn.LogSoftmax(dim=None) [исходный код]

Применяет функцию $\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.

Возвращает:

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

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

None

Примеры:

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

extra_repr() [исходный код]

Возвращает дополнительное представление модуля.

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

str

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

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

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

Tensor