Class Flatten

java.lang.Object
io.github.nearage.jnn.processing.Layer
io.github.nearage.jnn.model.layer.Flatten

public class Flatten extends Layer
TODO: Flatten layer
Author:
Nearage https://github.com/Nearage
  • Constructor Details

    • Flatten

      public Flatten()
      Initializes a flatten layer
  • Method Details

    • activate

      public Matrix activate(Matrix input) throws Exception
      Description copied from class: Layer
      Activates the layer for the given input
      Specified by:
      activate in class Layer
      Parameters:
      input - input data
      Returns:
      the activation of the layer
      Throws:
      Exception - a base exception with an error message
    • propagate

      public Matrix propagate(Matrix input) throws Exception
      Description copied from class: Layer
      Propagates the activation of the layer for the given input
      Specified by:
      propagate in class Layer
      Parameters:
      input - input data
      Returns:
      the propagation of the layer
      Throws:
      Exception - a base exception with an error message