torch.sparse.softmax

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

Применяет функцию softmax.

Softmax определяется как:

$\text{Softmax}(x_{i}) = \frac{exp(x_i)}{\sum_j exp(x_j)}$

где $i, j$ пробегают по индексам разреженного тензора, а неопределенные значения игнорируются. Это эквивалентно определению неопределенных значений как минус бесконечности, так что $exp(x_k) = 0$ когда запись с индексом $k$ не указана.

Она применяется ко всем срезам вдоль dim, и будет перемасштабировать их так, чтобы элементы лежали в диапазоне [0, 1] и суммировались до 1.

Параметры:

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