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。

引數

• input (張量) – 輸入張量

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

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