jax.nn.glu

jax.nn.glu(x, axis=-1)

閘門線性單元激活函數。

計算以下函數

\[\mathrm{glu}(x) = x\left[\ldots, 0:\frac{n}{2}, \ldots\right] \cdot \mathrm{sigmoid} \left( x\left[\ldots, \frac{n}{2}:n, \ldots\right] \right)\]

其中陣列沿著 axis 分成兩部分。axis 維度的大小必須可被二整除。

參數：

• x (ArrayLike) – 輸入陣列

• axis (int) – 應計算分割的軸 (預設值：-1)

傳回：

一個陣列。

傳回型別：

陣列

另請參閱

sigmoid()