jax.nn.glu

Contents

jax.nn.glu#

jax.nn.glu(x, axis=-1)[source]#

Gated linear unit activation function.

Parameters:

x (Any) – input array
axis (int) – the axis along which the split should be computed (default: -1)

Return type: