torch.sparse.softmax

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

应用一个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。

Parameters

• 输入 (张量) – 输入

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

• dtype (torch.dtype, 可选) – 返回张量所需的数据类型。如果指定，输入张量在操作执行前会被转换为dtype。这对于防止数据类型溢出很有用。默认值：None