Implement getOutputSize and getInputSize for seq layers

include/layers/input.cuh

@@ -13,25 +13,40 @@ class Input : public SequentialLayer {
   public:
     /**
      * @brief Create a new Input layer
-     * 
+     *
      * @param inputSize Size of the input vector
      */
     explicit Input(int inputSize);
 
     /**
      * @brief Destroy the Input layer
-     * 
+     *
      */
     ~Input();
 
     /**
-     * @brief Forward pass of the input layer. Just copies the input to the device
-     * 
+     * @brief Forward pass of the input layer. Just copies the input to the
+     * device
+     *
      * @param input Host pointer to the input vector
      * @return Device pointer to the output vector
      */
     float* forward(const float* input);
 
+    /**
+     * @brief Get output size
+     *
+     * @return int output size
+     */
+    int getOutputSize();
+
+        /**
+     * @brief Get input size
+     * 
+     * @return int input size
+     */
+    int getInputSize();
+
   private:
     int    inputSize;
     float* d_output;