diff --git a/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp b/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp index e92464e62c..622c030feb 100644 --- a/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp +++ b/src/mlpack/methods/ann/convolution_rules/naive_atrous_convolution.hpp @@ -111,8 +111,8 @@ class NaiveAtrousConvolution inputPadded.submat((filter.n_rows - 1)*dilation + filter.n_rows - 1, (filter.n_cols - 1)*dilation + filter.n_cols - 1, (filter.n_rows - 1)*dilation + filter.n_rows - 1 + input.n_rows - 1, - (filter.n_cols - 1)*dilation + filter.n_cols - 1 + input.n_cols - 1 - ) = input; + (filter.n_cols - 1)*dilation + filter.n_cols - 1 + input.n_cols - 1) + = input; NaiveAtrousConvolution::Convolution(inputPadded, filter, output, 1, 1, dilation); diff --git a/src/mlpack/methods/ann/layer/atrous_convolution.hpp b/src/mlpack/methods/ann/layer/atrous_convolution.hpp index 69bb3d9ffd..0848a23e7d 100644 --- a/src/mlpack/methods/ann/layer/atrous_convolution.hpp +++ b/src/mlpack/methods/ann/layer/atrous_convolution.hpp @@ -186,7 +186,7 @@ class AtrousConvolution * @param d The dilation size. * @return The convolution output size. */ - size_t ConvOutSize( const size_t size, + size_t ConvOutSize (const size_t size, const size_t k, const size_t s, const size_t p,