input shouldn't be sliced with expanded slices
This commit is contained in:
parent
56a1349959
commit
1d9c9644ee
1 changed files with 3 additions and 1 deletions
|
|
@ -479,7 +479,9 @@ def expand_wgmode_e(E: field_t,
|
|||
slices_exp[axis] = slice(E[0].shape[axis])
|
||||
slices_exp = (slice(3), *slices_exp)
|
||||
|
||||
Ee[slices_exp] = phase_E * numpy.array(E)[slices_Exp]
|
||||
slices_in = tuple(slice(3), *slices)
|
||||
|
||||
Ee[slices_exp] = phase_E * numpy.array(E)[slices_in]
|
||||
|
||||
return Ee
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue