Ignore size-1 axes when shifting

5 years ago · 6f2faca7dc
parent 63ad67800a
commit 6f2faca7dc
1 changed files with 2 additions and 0 deletions
--- a/meanas/fdfd/operators.py
+++ b/meanas/fdfd/operators.py
@ -531,6 +531,8 @@ def e_boundary_source(mask: vfield_t,
        shift = lambda axis, polarity: shift_with_mirror(axis=axis, shape=shape, shift_distance=polarity)

    for axis in (0, 1, 2):
+        if shape[axis] == 1:
+            continue
        for polarity in (-1, +1):
            r = shift(axis, polarity) - sparse.eye(numpy.prod(shape)) # shifted minus original
            r3 = sparse.block_diag((r, r, r))