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