提交 2dc20643 authored 作者: James Bergstra's avatar James Bergstra

added first module testcase! option to initialize Member on allocate

上级 e0e1eef8
......@@ -268,7 +268,7 @@ class Member(_RComponent):
r = self.r
if memo and r in memo:
return memo[r]
rval = gof.Container(r, storage = [None])
rval = gof.Container(r, storage = [getattr(r, 'data', None)])
memo[r] = io.In(result = r, value = rval, mutable = False)
return memo[r]
......
from theano.compile.module import *
import theano.tensor as T
def test_whats_up_with_submembers():
class Blah(FancyModule):
def __init__(self, stepsize):
super(Blah, self).__init__()
self.stepsize = Member(T.value(stepsize))
x = T.dscalar()
self.step = Method([x], x - self.stepsize)
B = Blah(0.0)
b = B.make(mode='FAST_RUN')
b.step(1.0)
print b.stepsize
assert b.stepsize == 0.0
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论