Implement getting layer, weights and biases

This commit is contained in:
2024-04-16 19:09:41 +02:00
parent f4ae45f867
commit 9fb9d7e8e1
7 changed files with 77 additions and 5 deletions

View File

@@ -2,6 +2,8 @@
#ifndef CUDANET_I_LAYER_H
#define CUDANET_I_LAYER_H
#include <vector>
namespace CUDANet::Layers {
/**
@@ -60,6 +62,12 @@ class WeightedLayer : public SequentialLayer {
*/
virtual void setWeights(const float* weights) = 0;
/**
* @brief Virtual function for getting weights
*
*/
virtual std::vector<float> getWeights() = 0;
/**
* @brief Virtual function for setting biases
*
@@ -67,6 +75,12 @@ class WeightedLayer : public SequentialLayer {
*/
virtual void setBiases(const float* biases) = 0;
/**
* @brief Virtual function for getting biases
*
*/
virtual std::vector<float> getBiases() = 0;
private:
/**
* @brief Initialize the weights