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$。

它会应用到所有沿 轴的切片上，并将其重新缩放，使元素值位于 [0, 1]范围内且总和为1。

参数

• 输入 (Tensor) – 输入张量

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

• dtype (torch.dtype, 可选) – 返回张量的数据类型。如果指定了 dtype，则在执行操作前将输入张量转换为指定的 dtype，以防止数据溢出。默认值：None