torch.nn.functional.softmax

torch.nn.functional.softmax(input, dim=None, _stacklevel=3, dtype=None) [исходный код]

Применяет функцию 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 (она работает быстрее и обладает лучшими численными свойствами).