torch.nn.functional.log_softmax

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

Применяет softmax, за которым следует логарифмирование.

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

См. LogSoftmax для получения дополнительных сведений.

Параметры:

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

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

Тензор