torch.nn.functional.softmax

torch.nn.functional.softmax(input, dim=None, _stacklevel=3, dtype=None) [source]

Применяет функцию softmax.

Softmax определяется как:

$\text{Softmax}(x_{i}) = \frac{\exp(x_i)}{\sum_j \exp(x_j)}$

Она применяется ко всем срезам вдоль dim и будет перемасштабировать их так, чтобы элементы находились в диапазоне [0, 1] и суммировались до 1.

См. Softmax для получения дополнительной информации.

Параметры

• input (Тензор) – входной тензор
• dim (int) – размерность, по которой будет вычисляться softmax.
• dtype (torch.dtype, необязательно) – желаемый тип данных возвращаемого тензора. Если указано, входной тензор приводится к типу dtype перед выполнением операции. Это полезно для предотвращения переполнения данных. По умолчанию: None.

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

Тензор

Примечание

Эта функция не работает напрямую с NLLLoss, которая ожидает, что логарифм будет вычислен между Softmax и собой. Используйте log_softmax вместо этого (это быстрее и имеет лучшие числовые свойства).