diff --git a/src/mlpack/methods/ann/layer/layer_norm.hpp b/src/mlpack/methods/ann/layer/layer_norm.hpp index 82587d137a..4305455010 100644 --- a/src/mlpack/methods/ann/layer/layer_norm.hpp +++ b/src/mlpack/methods/ann/layer/layer_norm.hpp @@ -133,7 +133,8 @@ class LayerNormType : public Layer // as the input. this->outputDimensions = this->inputDimensions; size = this->inputDimensions[0]; - for (size_t i=1; iinputDimensions.size(); i++) size *= this->inputDimensions[i]; + for (size_t i = 1; i < this->inputDimensions.size(); i++) + size *= this->inputDimensions[i]; } void SetWeights(typename MatType::elem_type* /* weightsPtr */) override;