jax-js
    Preparing search index...

    Function glu

    • Gated linear unit (GLU) activation function.

      Splits the axis dimension of the input into two halves, a and b, then computes a * sigmoid(b).

      Parameters

      Returns Array