jax.nn.selu

jax.nn.selu(x)

縮放指數線性單元激活。

計算逐元素函數

\[\begin{split}\mathrm{selu}(x) = \lambda \begin{cases} x, & x > 0\\ \alpha e^x - \alpha, & x \le 0 \end{cases}\end{split}\]

其中 \(\lambda = 1.0507009873554804934193349852946\) 和 \(\alpha = 1.6732632423543772848170429916717\)。

更多資訊，請參閱Self-Normalizing Neural Networks。

參數:

x (ArrayLike) – 輸入陣列

回傳:

一個陣列。

回傳型別:

陣列

另請參閱

elu()