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） 返回: 一个数组。 返回类型: Array 参见 sigmoid()
