提交 bbd1b11b authored 作者: James Bergstra's avatar James Bergstra

scan - added test of map-like usage

上级 1399eafc
from nose.plugins.skip import SkipTest
import unittest import unittest
import theano import theano
...@@ -373,6 +374,20 @@ class T_Scan(unittest.TestCase): ...@@ -373,6 +374,20 @@ class T_Scan(unittest.TestCase):
- optimization !? - optimization !?
''' '''
def test_map_functionality(self):
raise SkipTest('Map functionality not implemented yet')
def f_rnn(u_t):
return u_t + 3
u = theano.tensor.dvector()
Y, updts = theano.scan(f_rnn, sequences=u, outputs_taps={0:[]})
f2 = theano.function([u], Y, updates = updts)
v_u = numpy.array([1.,2.,3.,4.])
assert compareArrays(f2(v_u), v_u+3)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论