Layer.hpp

Go to the documentation of this file.

#pragma once
#include <Eigen/Core>
#include <Eigen/Dense>
#include "types.hpp"
#include "onnx/onnx_pb.h"
 
namespace mlask {
 
class Layer {
protected:
    std::size_t in_;
    std::size_t out_;
  public:
    std::size_t getIn()const{ return in_; }
    std::size_t getOut()const{ return out_; }
 
    virtual ~Layer() = default;
    virtual vectorOut forward(vectorIn) = 0;
    virtual vectorIn backward(vectorOut) = 0;
    virtual void fit(float_t learning_rate) = 0;
    virtual bool tryConvertToONNX(onnx::GraphProto* graph, std::string input, std::string output)const { return false; }
    virtual std::string str()const{ return "Layer"; }
};
 
template <typename T>
concept TLayer= std::derived_from<T, Layer>;
} // namespace mlask