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.

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

Тензор