Commit 06e2d7fa authored by Jait Dixit's avatar Jait Dixit
Browse files

Fix get_centering_mask calls

parent 38aa1c4b
......@@ -451,11 +451,13 @@ class FFTWTransformInfo(object):
self.cmask_domain = fftw_context.get_centering_mask(
domain.zerocenter,
local_shape if axes is None else
[y for x, y in enumerate(local_shape) if x in axes],
local_offset_Q)
self.cmask_codomain = fftw_context.get_centering_mask(
codomain.zerocenter,
local_shape if axes is None else
[y for x, y in enumerate(local_shape) if x in axes],
local_offset_Q)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment