Use arma::pow to use fast armadillo computation

2021-06-21 12:10:20 +05:30
parent 821a2f2712
commit 83207c4439
1 changed files with 2 additions and 1 deletions
@@ -195,9 +195,10 @@ class LpPooling
        if (colidx >= 1)
          val -= inputPre(rowEnd, colidx - 1);

-        output(i, j) = pow(val, 1.0 / normType);
+        output(i, j) = val;
      }
    }
+    output = arma::pow(output, 1.0 / normType);
  }

  /**