torch.nn.functional.log_softmax

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

Применяет softmax, а затем логарифм.

Хотя математически это эквивалентно log(softmax(x)), выполнение этих двух операций по отдельности происходит медленнее и приводит к численной нестабильности. Эта функция использует альтернативную формулу для корректного вычисления результата и градиента.

Подробнее см. в разделе LogSoftmax.

Параметры:

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

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

Tensor