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

Возвращаемый тип:

Tensor

Примечание

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