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
065ec7ac
Commit
065ec7ac
authored
Aug 25, 2019
by
Philipp Arras
Browse files
Simplify for non_constant input to Hamiltonian
parent
12f34692
Changes
1
Hide whitespace changes
Inline
Side-by-side
nifty5/operators/energy_operators.py
View file @
065ec7ac
...
...
@@ -319,9 +319,11 @@ class StandardHamiltonian(EnergyOperator):
`<https://arxiv.org/abs/1812.04403>`_
"""
def
__init__
(
self
,
lh
,
ic_samp
=
None
):
def
__init__
(
self
,
lh
,
ic_samp
=
None
,
_c_inp
=
None
):
self
.
_lh
=
lh
self
.
_prior
=
GaussianEnergy
(
domain
=
lh
.
domain
)
if
_c_inp
is
not
None
:
_
,
self
.
_prior
=
self
.
_prior
.
simplify_for_constant_input
(
_c_inp
)
self
.
_ic_samp
=
ic_samp
self
.
_domain
=
lh
.
domain
...
...
@@ -338,9 +340,13 @@ class StandardHamiltonian(EnergyOperator):
def
__repr__
(
self
):
subs
=
'Likelihood:
\n
{}'
.
format
(
utilities
.
indent
(
self
.
_lh
.
__repr__
()))
subs
+=
'
\n
Prior:
Quadratic
{}'
.
format
(
self
.
_
lh
.
domain
.
keys
()
)
subs
+=
'
\n
Prior:
\n
{}'
.
format
(
self
.
_
prior
)
return
'StandardHamiltonian:
\n
'
+
utilities
.
indent
(
subs
)
def
_simplify_for_constant_input_nontrivial
(
self
,
c_inp
):
out
,
lh1
=
self
.
_lh
.
simplify_for_constant_input
(
c_inp
)
return
out
,
StandardHamiltonian
(
lh1
,
self
.
_ic_samp
,
_c_inp
=
c_inp
)
class
AveragedEnergy
(
EnergyOperator
):
"""Averages an energy over samples.
...
...
Write
Preview
Markdown
is supported
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