From 99348fcea717d23ff4e100bf9763a874ad4e033d Mon Sep 17 00:00:00 2001 From: Adam Kropp <30504149+akropp@users.noreply.github.com> Date: Wed, 1 Nov 2023 18:11:33 -0400 Subject: [PATCH] Update src/mlpack/methods/ann/layer/layer_norm.hpp Co-authored-by: Ryan Curtin --- src/mlpack/methods/ann/layer/layer_norm.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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;