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