brainpy.math.activations.glu#

brainpy.math.activations.glu(x, axis=- 1)[source]#

Gated linear unit activation function.

Parameters

  • x (JaxArray, jnp.ndarray) – The input array.

  • axis (int) – The axis along which the split should be computed (default: -1)