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