torch.sparse.log_softmax

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

套用 softmax 函數，然後取對數。

詳情請參閱 softmax。

參數

• input (Tensor) – 輸入

• dim (int) – softmax 將沿著此維度計算。

• dtype (torch.dtype, optional) – 返回 tensor 的目標數據類型。 如果指定，輸入 tensor 將在執行操作之前被轉換為 dtype。 這對於防止數據類型溢位很有用。 預設值：None