Class LinearLayer

java.lang.Object
smile.deep.layer.LinearLayer
All Implemented Interfaces:
Function<Tensor,Tensor>, Layer

public class LinearLayer extends Object implements Layer
A fully connected linear layer.
  • Constructor Details

    • LinearLayer

      public LinearLayer(int in, int out)
      Constructor.
      Parameters:
      in - the number of input features.
      out - the number of output features.
    • LinearLayer

      public LinearLayer(int in, int out, boolean bias)
      Constructor.
      Parameters:
      in - the number of input features.
      out - the number of output features.
      bias - If false, the layer will not learn an additive bias.
  • Method Details

    • asTorch

      public org.bytedeco.pytorch.Module asTorch()
      Description copied from interface: Layer
      Returns the PyTorch Module object.
      Specified by:
      asTorch in interface Layer
      Returns:
      the PyTorch Module object.
    • forward

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