Class LogSoftmax

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

public class LogSoftmax extends ActivationFunction
Log softmax activation function.
See Also:
  • Constructor Details

    • LogSoftmax

      public LogSoftmax()
      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.