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$ 运行稀疏张量的索引，未指定项被忽略。这等同于将未指定项定义为负无穷大，因此当索引为 $k$ 的项未指定时，$exp(x_k) = 0$。

它被应用于 dim 上的所有切片，并将它们重新缩放，使得元素位于 [0, 1] 范围内且总和为 1。

参数

• input (张量) – 输入

• dim (int) – 将计算 softmax 的维度。

• dtype (torch.dtype, 可选) – 返回张表的期望数据类型。如果指定，则在执行操作之前，输入张表将被转换为 dtype。这有助于防止数据类型溢出。默认值：None