Python PyTorch log_softmax用法及代碼示例

本文簡要介紹python語言中 torch.special.log_softmax 的用法。

用法:

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

參數：

• input(Tensor) -輸入

• dim(int) -將沿其計算log_softmax 的維度。

• dtype(torch.dtype， 可選的) -返回張量的所需數據類型。如果指定，則在執行操作之前將輸入張量強製轉換為dtype。這對於防止數據類型溢出很有用。默認值：無。

計算 softmax，後跟對數。

雖然在數學上等價於 log(softmax(x))，但單獨執行這兩個操作更慢且數值不穩定。該函數計算為：

\text{log\_softmax}(x_{i}) = \log\left(\frac{\exp(x_i) }{ \sum_j \exp(x_j)} \right)

例子：

>>> t = torch.ones(2, 2)
>>> torch.special.log_softmax(t, 0)
tensor([[-0.6931, -0.6931],
        [-0.6931, -0.6931]])