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。

引數

• input (張量) – 輸入

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

• dtype (torch.dtype, optional) – 返回張表的期望資料型別。如果指定，在執行操作之前，輸入張表將被轉換為 dtype。這對於防止資料型別溢位很有用。預設為 None。