flax.linen.glu#

flax.linen.glu(x, axis=- 1)[source]#

Gated linear unit activation function.

Parameters

  • x – input array

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