Fix shift_with_mirror() for C-ordered arrays

5 years ago · 0503e9d6ef
parent b466ed02ea
commit 0503e9d6ef
1 changed files with 1 additions and 3 deletions
--- a/meanas/fdfd/operators.py
+++ b/meanas/fdfd/operators.py
@ -341,9 +341,7 @@ def shift_with_mirror(axis: int, shape: List[int], shift_distance: int=1) -> spa

    n = numpy.prod(shape)
    i_ind = numpy.arange(n)
-    j_ind = ijk[0] + ijk[1] * shape[0]
-    if len(shape) == 3:
-        j_ind += ijk[2] * shape[0] * shape[1]
+    j_ind = numpy.ravel_multi_index(ijk, shape, order='C')

    vij = (numpy.ones(n), (i_ind, j_ind.ravel(order='C')))