Initial model implementation

This commit is contained in:
2024-03-20 22:31:39 +01:00
parent 6f4cdf3792
commit af6838e8ae
4 changed files with 76 additions and 2 deletions

View File

@@ -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