Skip recurssion for flat matrices

josdejong · Feb 12, 2025 · 105759c · 105759c
1 parent 9727495
commit 105759c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/type/matrix/DenseMatrix.js b/src/type/matrix/DenseMatrix.js
@@ -600,8 +600,18 @@ export const createDenseMatrixClass = /* #__PURE__ */ factory(name, dependencies
    */
   DenseMatrix.prototype.map = function (callback) {
     const me = this
+    const size = me._size
+    const maxDepth = size.length
+
+    if (maxDepth === 0) return me.clone()
+
     const result = new DenseMatrix(me)
     const fastCallback = optimizeCallback(callback, me._data, 'map')
+    if (maxDepth === 1) {
+      return new DenseMatrix(
+        me._data.map((value, index) => fastCallback(value, [index], me))
+      )
+    }
 
     result._forEach(function (arr, i, index) {
       arr[i] = fastCallback(arr[i], index, me)