layer.h

Go to the documentation of this file.

#ifndef LAYER_H
#define LAYER_H

#include <vector>
#include <iostream>
#include <algorithm>
#include <random>
#include "linalg.h"

class Layer {
    public:
        Layer(int _input_unit, int _output_unit);
        inline friend std::ostream& operator<< (std::ostream &out, const Layer & layer) {
            out << "_nodes is: " << std::endl;
            out << *layer._nodes<< std::endl;
            out << "_bias is: " << std::endl;
            out << *layer._bias << std::endl;
            return out;
        }
        matrix* _nodes;
        vector* _bias;
    private:
        int _input_unit;
        int _output_unit;
        int _init_layer();
};

#endif