torch.sparse.log_softmax

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

應用 softmax 函式後取對數。

更多細節請參閱 softmax。

引數

• input (張量) – 輸入

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

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