apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update src/mlpack/methods/ann/layer/layer_norm.hpp

Browse Source 
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
Adam Kropp
2023-11-01 18:11:33 -04:00
committed by GitHub
co-authored by Ryan Curtin
parent 103db420b9
commit 99348fcea7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/layer_norm.hpp
+2 -1
View File
@@ -133,7 +133,8 @@ class LayerNormType : public Layer<MatType>
// as the input.
this->outputDimensions = this->inputDimensions;
size = this->inputDimensions[0];
for (size_t i=1; i<this->inputDimensions.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;