mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
30 lines
481 B
Plaintext
30 lines
481 B
Plaintext
#ifndef INPUT_LAYER_H
|
|
#define INPUT_LAYER_H
|
|
|
|
#include <ilayer.cuh>
|
|
|
|
namespace Layers {
|
|
|
|
class Input : public ILayer {
|
|
public:
|
|
Input(int inputSize);
|
|
~Input();
|
|
|
|
float* forward(const float* input);
|
|
|
|
void setWeights(const float* weights);
|
|
void setBiases(const float* biases);
|
|
|
|
private:
|
|
void initializeWeights();
|
|
void initializeBiases();
|
|
|
|
void toCuda();
|
|
|
|
int inputSize;
|
|
float* d_output;
|
|
};
|
|
|
|
} // namespace Layers
|
|
|
|
#endif // INPUT_LAYER_H |