Class GLU

All Implemented Interfaces:
Serializable, Function<Tensor,Tensor>, Layer

public class GLU extends ActivationFunction
Gated Linear Unit activation function.
See Also:
  • Constructor Details

    • GLU

      public GLU()
      Constructor.
  • Method Details

    • forward

      public Tensor forward(Tensor x)
      Description copied from interface: Layer
      Forward propagation (or forward pass) through the layer.
      Parameters:
      x - the input tensor.
      Returns:
      the output tensor.