torch.sparse.log_softmax

torch.sparse.log_softmax(input, dim, *, dtype=None) → Tensor

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

См. softmax для получения более подробной информации.

Параметры

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