flatten_layer.h

Go to the documentation of this file.

 
#pragma once
#ifndef FLATTEN_LAYER_H 
#define FLATTEN_LAYER_H
 
#include <iostream>
#include <eigen3/Eigen/Dense>
#include <EigenRand/EigenRand>
#include "layer.h"
#include <chrono>
 
namespace Neural
{
    class Flatten_Layer : public Layer
    {
 
        public:
            Flatten_Layer();
            virtual Eigen::MatrixXd Forward_propagation(Eigen::MatrixXd input);
            virtual Eigen::MatrixXd Backward_propagation(Eigen::MatrixXd output_error, float learning_rate);
            virtual Json::Value toJSON();
 
 
        protected:
 
    };
}
#endif