non-copying operation to improve speed for large matrices (useful for…

… 3D simulation) (#269)
waltsims · Jan 18, 2024 · e31e0ed · e31e0ed
1 parent 9f57fbb
commit e31e0ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kwave/utils/matlab.py b/kwave/utils/matlab.py
@@ -47,7 +47,7 @@ def matlab_assign(matrix: np.ndarray, indices: Union[int, np.ndarray],
 
     """
     original_shape = matrix.shape
-    matrix = matrix.flatten(order='F')
+    matrix = np.ravel(matrix, order='F')
     matrix[indices] = values
     return matrix.reshape(original_shape, order='F')