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。

参数:

• 输入（张量）- 输入

• dim（整数）- softmax 将计算的维度。

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