netket.nn.glu

netket.nn.glu(x, axis=- 1)

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