torch.sparse.log_softmax

torch.sparse.log_softmax(input, dim, *, dtype=None) → Tensor

應用 Softmax 函式後接對數運算。

更多詳情請參見 softmax。

引數

• input (Tensor) – 輸入

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

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