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 (Tensor) – 輸入

• dim (int) – 計算 softmax 的維度。

• dtype (torch.dtype, optional) – 返回張量所需之資料型別。 如果指定，則輸入張量在執行操作之前會被轉換為 dtype。 這有助於防止資料型別溢位。 預設值：None