__init__.py 4.73 KB
Newer Older
1
from .version import __version__
Martin Reinecke's avatar
Martin Reinecke committed
2
3

from . import dobj
Martin Reinecke's avatar
Martin Reinecke committed
4
5
6
7
8
9
10
11
12
13
14
15

from .domains.domain import Domain
from .domains.structured_domain import StructuredDomain
from .domains.unstructured_domain import UnstructuredDomain
from .domains.rg_space import RGSpace
from .domains.lm_space import LMSpace
from .domains.gl_space import GLSpace
from .domains.hp_space import HPSpace
from .domains.power_space import PowerSpace
from .domains.dof_space import DOFSpace
from .domains.log_rg_space import LogRGSpace

16
from .domain_tuple import DomainTuple
17
from .field import Field
Philipp Arras's avatar
Fixups  
Philipp Arras committed
18
19
20

from .nonlinearities import Exponential, Linear, PositiveTanh, Tanh

Martin Reinecke's avatar
Martin Reinecke committed
21
22
23
from .models.constant import Constant
from .models.linear_model import LinearModel
from .models.local_nonlinearity import (LocalModel, PointwiseExponential,
Martin Reinecke's avatar
Martin Reinecke committed
24
                                        PointwisePositiveTanh, PointwiseTanh)
Martin Reinecke's avatar
Martin Reinecke committed
25
26
27
28
from .models.model import Model
from .models.multi_model import MultiModel
from .models.variable import Variable

29
from .operators.central_zero_padder import CentralZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
30
31
32
33
34
35
from .operators.diagonal_operator import DiagonalOperator
from .operators.dof_distributor import DOFDistributor
from .operators.domain_distributor import DomainDistributor
from .operators.endomorphic_operator import EndomorphicOperator
from .operators.exp_transform import ExpTransform
from .operators.fft_operator import FFTOperator
Martin Reinecke's avatar
Martin Reinecke committed
36
from .operators.field_adapter import FieldAdapter
Martin Reinecke's avatar
Martin Reinecke committed
37
from .operators.field_zero_padder import FieldZeroPadder
Martin Reinecke's avatar
Martin Reinecke committed
38
39
from .operators.hartley_operator import HartleyOperator
from .operators.harmonic_smoothing_operator import HarmonicSmoothingOperator
Martin Reinecke's avatar
Martin Reinecke committed
40
41
42
43
44
45
46
from .operators.geometry_remover import GeometryRemover
from .operators.harmonic_transform_operator import HarmonicTransformOperator
from .operators.inversion_enabler import InversionEnabler
from .operators.laplace_operator import LaplaceOperator
from .operators.linear_operator import LinearOperator
from .operators.mask_operator import MaskOperator
from .operators.multi_adaptor import MultiAdaptor
47
from .operators.null_operator import NullOperator
Martin Reinecke's avatar
Martin Reinecke committed
48
49
50
51
52
53
54
55
56
from .operators.power_distributor import PowerDistributor
from .operators.qht_operator import QHTOperator
from .operators.sampling_enabler import SamplingEnabler
from .operators.sandwich_operator import SandwichOperator
from .operators.scaling_operator import ScalingOperator
from .operators.selection_operator import SelectionOperator
from .operators.slope_operator import SlopeOperator
from .operators.smoothness_operator import SmoothnessOperator
from .operators.symmetrizing_operator import SymmetrizingOperator
Martin Reinecke's avatar
Martin Reinecke committed
57
from .operators.vdot_operator import VdotOperator
Martin Reinecke's avatar
Martin Reinecke committed
58

59
60
from .probing.utils import probe_with_posterior_samples, probe_diagonal, \
    StatCalculator
Martin Reinecke's avatar
Martin Reinecke committed
61
62
63
64
65
66
67
68
69
70
71
72
73

from .minimization.line_search import LineSearch
from .minimization.line_search_strong_wolfe import LineSearchStrongWolfe
from .minimization.iteration_controller import IterationController
from .minimization.gradient_norm_controller import GradientNormController
from .minimization.minimizer import Minimizer
from .minimization.conjugate_gradient import ConjugateGradient
from .minimization.nonlinear_cg import NonlinearCG
from .minimization.descent_minimizer import DescentMinimizer
from .minimization.steepest_descent import SteepestDescent
from .minimization.vl_bfgs import VL_BFGS
from .minimization.l_bfgs import L_BFGS
from .minimization.relaxed_newton import RelaxedNewton
Martin Reinecke's avatar
Martin Reinecke committed
74
75
from .minimization.scipy_minimizer import (ScipyMinimizer, NewtonCG, L_BFGS_B,
                                           ScipyCG)
Martin Reinecke's avatar
Martin Reinecke committed
76
77
78
79
from .minimization.energy import Energy
from .minimization.quadratic_energy import QuadraticEnergy
from .minimization.line_energy import LineEnergy
from .minimization.energy_sum import EnergySum
Martin Reinecke's avatar
Martin Reinecke committed
80

Martin Reinecke's avatar
Martin Reinecke committed
81
from .sugar import *
Martin Reinecke's avatar
Martin Reinecke committed
82
from .plotting.plot import plot, plot_finish
Philipp Arras's avatar
Fixups  
Philipp Arras committed
83

Martin Reinecke's avatar
Martin Reinecke committed
84
85
86
87
88
from .library.amplitude_model import make_amplitude_model
from .library.gaussian_energy import GaussianEnergy
from .library.los_response import LOSResponse
from .library.point_sources import PointSources
from .library.poissonian_energy import PoissonianEnergy
Philipp Arras's avatar
Fixup  
Philipp Arras committed
89
from .library.wiener_filter_curvature import WienerFilterCurvature
Martin Reinecke's avatar
Martin Reinecke committed
90
91
from .library.correlated_fields import (make_correlated_field,
                                        make_mf_correlated_field)
Jakob Knollmueller's avatar
Jakob Knollmueller committed
92
from .library.bernoulli_energy import BernoulliEnergy
Martin Reinecke's avatar
Martin Reinecke committed
93

Philipp Arras's avatar
Move it  
Philipp Arras committed
94
from . import extra
95

Martin Reinecke's avatar
Martin Reinecke committed
96
from .utilities import memo, frozendict
Martin Reinecke's avatar
Martin Reinecke committed
97

Martin Reinecke's avatar
Martin Reinecke committed
98
99
from .logger import logger

Martin Reinecke's avatar
Martin Reinecke committed
100
101
102
from .multi.multi_domain import MultiDomain
from .multi.multi_field import MultiField
from .multi.block_diagonal_operator import BlockDiagonalOperator
Martin Reinecke's avatar
Martin Reinecke committed
103

Martin Reinecke's avatar
Martin Reinecke committed
104
105
from .energies.kl import SampledKullbachLeiblerDivergence
from .energies.hamiltonian import Hamiltonian
106

Martin Reinecke's avatar
Martin Reinecke committed
107
108
from.operator import Linearization, Operator

109
# We deliberately don't set __all__ here, because we don't want people to do a
Philipp Arras's avatar
Philipp Arras committed
110
# "from nifty5 import *"; that would swamp the global namespace.