fixed input layer

Dpbm · Jan 28, 2024 · 2ec7e76 · 2ec7e76
1 parent 0ba5537
commit 2ec7e76
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/machine/layer.cpp b/machine/layer.cpp
@@ -4,16 +4,15 @@
 using Matrices::Matrix;
 
 namespace Layers {
-  unsigned int width, height;
+  unsigned int size;
   Matrix *values;
 
-  InputLayer::InputLayer(unsigned int width, unsigned int height){
-    this->width = width;
-    this->height = height;
+  InputLayer::InputLayer(unsigned int size){
+    this->size = size;
   }
 
   void InputLayer::set_node_values(Matrix *values){
-    if(values->get_width() != this->width || values->get_height() != this->height)
+    if(values->get_width() != 1 || values->get_height() != this->size)
       throw std::invalid_argument("Invalid values dimensions!");
 
     this->values = values;

diff --git a/machine/layer.h b/machine/layer.h
@@ -12,11 +12,11 @@ namespace Layers{
 
   class InputLayer: Layer{
     public:
-      InputLayer(unsigned int width, unsigned int height);
+      InputLayer(unsigned int size);
       void set_node_values(Matrix *values);
       ~InputLayer();
     private:
-      unsigned int width, height;
+      unsigned int size;
       Matrix *values;
   };