mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 18:24:26 +00:00
Implement getOutputSize and getInputSize for seq layers
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user