Implement getOutputSize and getInputSize for seq layers

This commit is contained in:
2024-04-22 20:31:58 +02:00
parent fbf6c44bdd
commit f17debc244
18 changed files with 186 additions and 66 deletions

View File

@@ -80,13 +80,18 @@ class Conv2d : public WeightedLayer {
std::vector<float> getBiases();
/**
* @brief Get the output width (/ height) of the layer
* @brief Get output size
*
* @return int
* @return int output size
*/
int getOutputSize() {
return outputSize;
}
int getOutputSize();
/**
* @brief Get input size
*
* @return int input size
*/
int getInputSize();
/**
* @brief Get the padding size of the layer