diff --git a/mlpp/activation/activation.cpp b/mlpp/activation/activation.cpp index 6be0b6f..cc6f3cf 100644 --- a/mlpp/activation/activation.cpp +++ b/mlpp/activation/activation.cpp @@ -858,7 +858,7 @@ Ref MLPPActivation::sigmoid_normv(const Ref &z) { } Ref MLPPActivation::sigmoid_normm(const Ref &z) { MLPPLinAlg alg; - return alg.division_element_wisenvnm(alg.onematnm(z->size().x, z->size().y), alg.additionnm(alg.onematnm(z->size().x, z->size().y), alg.expnm(alg.scalar_multiplynm(-1, z)))); + return alg.division_element_wisenvnm(alg.onematnm(z->size().y, z->size().x), alg.additionnm(alg.onematnm(z->size().y, z->size().x), alg.expnm(alg.scalar_multiplynm(-1, z)))); } real_t MLPPActivation::sigmoid_derivr(real_t z) {