Layer.hpp

Go to the documentation of this file.

#pragma once
 
#include <Eigen/Dense>
#include "../Utilities/clue.hpp"
 
namespace NNFS
{
    enum class LayerType
    {
        DENSE,
        ACTIVATION
    };
 
    class Layer
    {
    public:
        LayerType type; // Type of layer
 
    public:
        Layer(LayerType type) : type(type) {}
 
        virtual ~Layer() = default;
 
        virtual void forward(Eigen::MatrixXd &out, const Eigen::MatrixXd &x) = 0;
 
        virtual void backward(Eigen::MatrixXd &out, const Eigen::MatrixXd &dx) = 0;
    };
} // namespace NNFS