jax.nn.elu#

jax.nn.elu(x, alpha=1.0)[原始碼]#

指數線性單元激活函數。

計算逐元素函數

\[\begin{split}\mathrm{elu}(x) = \begin{cases} x, & x > 0\\ \alpha \left(\exp(x) - 1\right), & x \le 0 \end{cases}\end{split}\]
參數:
  • x (ArrayLike) – 輸入陣列

  • alpha (ArrayLike) – alpha 值純量或陣列 (預設:1.0)

回傳:

一個陣列。

回傳型別:

Array

另請參閱

selu()