mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Implement getting layer, weights and biases
This commit is contained in:
@@ -43,6 +43,13 @@ class Dense : public WeightedLayer {
|
||||
*/
|
||||
void setWeights(const float* weights);
|
||||
|
||||
/**
|
||||
* @brief Get the weights of the layer
|
||||
*
|
||||
* @return Vector of weights
|
||||
*/
|
||||
std::vector<float> getWeights();
|
||||
|
||||
/**
|
||||
* @brief Set the biases of the layer
|
||||
*
|
||||
@@ -50,6 +57,13 @@ class Dense : public WeightedLayer {
|
||||
*/
|
||||
void setBiases(const float* biases);
|
||||
|
||||
/**
|
||||
* @brief Get the biases of the layer
|
||||
*
|
||||
* @return Vector of biases
|
||||
*/
|
||||
std::vector<float> getBiases();
|
||||
|
||||
private:
|
||||
unsigned int inputSize;
|
||||
unsigned int outputSize;
|
||||
|
||||
Reference in New Issue
Block a user