Skip to content
GitLab
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
7db7646b
Commit
7db7646b
authored
Jun 15, 2018
by
Philipp Arras
Browse files
Add explicit local models
parent
93314cc2
Pipeline
#31078
passed with stages
in 1 minute and 24 seconds
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
nifty4/models/__init__.py
View file @
7db7646b
from
.constant
import
Constant
from
.local_nonlinearity
import
LocalModel
from
.local_nonlinearity
import
(
LocalModel
,
PointwiseExponential
,
PointwisePositiveTanh
,
PointwiseTanh
)
from
.model
import
LinearModel
,
Model
from
.variable
import
Variable
__all__
=
[
'Model'
,
'Constant'
,
'LocalModel'
,
'Variable'
,
'LinearModel'
]
'LinearModel'
,
'PointwiseTanh'
,
'PointwisePositiveTanh'
,
'PointwiseExponential'
]
nifty4/models/local_nonlinearity.py
View file @
7db7646b
from
nifty4.library.nonlinearities
import
Exponential
,
PositiveTanh
,
Tanh
from
nifty4.sugar
import
makeOp
from
.model
import
Model
...
...
@@ -18,3 +19,15 @@ class LocalModel(Model):
def
at
(
self
,
position
):
return
self
.
__class__
(
self
.
_inp
.
at
(
position
),
self
.
_nonlinearity
)
def
PointwiseExponential
(
inp
):
return
LocalModel
(
inp
,
Exponential
())
def
PointwiseTanh
(
inp
):
return
LocalModel
(
inp
,
Tanh
())
def
PointwisePositiveTanh
(
inp
):
return
LocalModel
(
inp
,
PositiveTanh
())
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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