torch.sparse.log_softmax

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

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

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

Параметры:

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