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

@@ -58,6 +58,13 @@ class Conv2d : public WeightedLayer {
*/
void setWeights(const float* weights_input);
/**
* @brief Get the weights of the convolutional layer
*
* @return std::vector<float>
*/
std::vector<float> getWeights();
/**
* @brief Set the biases of the convolutional layer
*
@@ -65,6 +72,13 @@ class Conv2d : public WeightedLayer {
*/
void setBiases(const float* biases_input);
/**
* @brief Get the biases of the convolutional layer
*
* @return std::vector<float>
*/
std::vector<float> getBiases();
/**
* @brief Get the output width (/ height) of the layer
*