jax.nn.glu

Warning

This page was created from a pull request (#9655).

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

Any

previous

jax.nn.gelu

next

jax.nn.softmax