Initial model implementation

include/layers/input.cuh

@@ -1,20 +1,22 @@
 #ifndef CUDANET_INPUT_LAYER_H
 #define CUDANET_INPUT_LAYER_H
 
+#include "layer.cuh"
+
 namespace CUDANet::Layers {
 
 /**
  * @brief Input layer, just copies the input to the device
  *
  */
-class Input {
+class Input : public SequentialLayer {
   public:
     /**
      * @brief Create a new Input layer
      * 
      * @param inputSize Size of the input vector
      */
-    Input(int inputSize);
+    explicit Input(int inputSize);
 
     /**
      * @brief Destroy the Input layer

include/model/model.hpp

@@ -0,0 +1,34 @@
+#ifndef CUDANET_MODEL_H
+#define CUDANET_MODEL_H
+
+#include <string>
+#include <vector>
+#include <map>
+#include "layer.cuh"
+
+namespace CUDANet {
+
+class Model {
+  public:
+    Model(const int inputSize, const int inputChannels);
+    ~Model();
+
+    float* predict(const float* input);
+
+    void addLayer(const std::string& name, Layers::SequentialLayer* layer);
+
+  private:
+
+    int inputSize;
+    int inputChannels;
+
+    int outputSize;
+    
+    std::vector<Layers::SequentialLayer*> layers;
+    std::map<std::string, Layers::WeightedLayer*> layerMap;
+
+};
+
+}  // namespace CUDANet
+
+#endif  // CUDANET_MODEL_H