Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
ift
NIFTy
Commits
8d067d40
Commit
8d067d40
authored
Sep 13, 2017
by
Martin Reinecke
Browse files
PEP8
parent
deeff6f9
Pipeline
#18168
passed with stage
in 5 minutes and 28 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
nifty2go/operators/fft_operator/fft_operator_support.py
View file @
8d067d40
...
...
@@ -51,9 +51,9 @@ class RGRGTransformation(Transformation):
raise
ValueError
(
"Provided axes does not match array shape"
)
from
pyfftw.interfaces.numpy_fft
import
rfftn
if
issubclass
(
a
.
dtype
.
type
,
np
.
complexfloating
):
raise
TypeError
(
"Hartley tansform
works for
real-valued arrays
only
."
)
tmp
=
rfftn
(
a
,
axes
=
axes
)
if
issubclass
(
a
.
dtype
.
type
,
np
.
complexfloating
):
raise
TypeError
(
"Hartley tansform
requires
real-valued arrays."
)
tmp
=
rfftn
(
a
,
axes
=
axes
)
res
=
np
.
empty_like
(
a
)
if
axes
is
None
:
axes
=
list
(
range
(
a
.
ndim
))
...
...
@@ -62,15 +62,14 @@ class RGRGTransformation(Transformation):
ntmplast
=
tmp
.
shape
[
lastaxis
]
nrem
=
nlast
-
ntmplast
slice1
=
[
slice
(
None
)]
*
a
.
ndim
slice1
[
lastaxis
]
=
slice
(
0
,
ntmplast
)
slice1
[
lastaxis
]
=
slice
(
0
,
ntmplast
)
res
[
slice1
]
=
tmp
.
real
+
tmp
.
imag
tmp
=
np
.
roll
(
tmp
,
-
1
,
axes
)
slice1
=
[
slice
(
None
)]
*
a
.
ndim
slice1
[
lastaxis
]
=
slice
(
ntmplast
,
None
)
tmp
=
np
.
roll
(
tmp
,
-
1
,
axes
)
slice1
[
lastaxis
]
=
slice
(
ntmplast
,
None
)
slice2
=
[
slice
(
None
)]
*
a
.
ndim
for
i
in
axes
:
slice2
[
i
]
=
slice
(
None
,
None
,
-
1
)
slice2
[
lastaxis
]
=
slice
(
nrem
-
1
,
None
,
-
1
)
slice2
[
i
]
=
slice
(
None
,
None
,
-
1
)
slice2
[
lastaxis
]
=
slice
(
nrem
-
1
,
None
,
-
1
)
res
[
slice1
]
=
tmp
[
slice2
].
real
-
tmp
[
slice2
].
imag
return
res
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment