當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python mxnet.symbol.op.log_softmax用法及代碼示例


用法:

mxnet.symbol.op.log_softmax(data=None, axis=_Null, temperature=_Null, dtype=_Null, use_length=_Null, name=None, attr=None, out=None, **kwargs)

參數

  • data(Symbol) - 輸入數組。
  • axis(int, optional, default='-1') - 計算 softmax 的軸。
  • temperature(double or None, optional, default=None) - softmax 中的溫度參數
  • dtype({None, 'float16', 'float32', 'float64'},optional, default='None') - 輸出的 DType,以防無法推斷。如果未定義(dtype=None),則默認與輸入的 dtype 相同。
  • use_length(boolean or None, optional, default=0) - 是否使用長度輸入作為數據輸入的掩碼。
  • name(string, optional.) - 結果符號的名稱。

返回

結果符號。

返回類型

Symbol

計算輸入的 log softmax。這相當於計算 softmax 後跟 log。

例子:

>>> x = mx.nd.array([1, 2, .1])
>>> mx.nd.log_softmax(x).asnumpy()
array([-1.41702998, -0.41702995, -2.31702995], dtype=float32)

>>> x = mx.nd.array( [[1, 2, .1],[.1, 2, 1]] )
>>> mx.nd.log_softmax(x, axis=0).asnumpy()
array([[-0.34115392, -0.69314718, -1.24115396],
       [-1.24115396, -0.69314718, -0.34115392]], dtype=float32)

相關用法


注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.symbol.op.log_softmax。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。