summaryrefslogtreecommitdiff
path: root/numpy/dual.py
blob: 32baa020eba022645a502aa595bf781360daec06 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# This module should be used for functions both in numpy and scipy if
#  you want to use the numpy version if available but the scipy version
#  otherwise.
#  Usage  --- from numpy.dual import fft, inv

__all__ = ['fft','ifft','fftn','ifftn','fft2','ifft2',
           'inv','svd','solve','det','eig','eigvals','lstsq',
           'pinv','cholesky','i0']

try:
    import scipy.linalg as linpkg
except ImportError:
    import numpy.linalg as linpkg

try:
    import scipy.fftpack as fftpkg
except ImportError:
    import numpy.dft as fftpkg

try:
    from scipy.special import i0
except ImportError:
    from numpy.lib import i0

fft = fftpkg.fft
ifft = fftpkg.ifft
fftn = fftpkg.fftn
ifftn = fftpkg.ifftn
fft2 = fftpkg.fft2
ifft2 = fftpkg.ifft2

inv = linpkg.inv
svd = linpkg.svd
solve = linpkg.solve
det = linpkg.det
eig = linpkg.eig
eigvals = linpkg.eigvals
lstsq = linpkg.lstsq
pinv = linpkg.pinv
cholesky = linpkg.cholesky