提交 20cc80db authored 作者: Kyunghyun Cho's avatar Kyunghyun Cho

added a unittest for failure mode

上级 26477f9f
...@@ -11,6 +11,7 @@ import numpy ...@@ -11,6 +11,7 @@ import numpy
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
from nose.plugins.attrib import attr from nose.plugins.attrib import attr
from nose.tools import assert_raises from nose.tools import assert_raises
from nose.tools import raises
from numpy.testing import dec from numpy.testing import dec
import theano import theano
...@@ -3932,6 +3933,27 @@ class T_Scan(unittest.TestCase): ...@@ -3932,6 +3933,27 @@ class T_Scan(unittest.TestCase):
assert diff <= type_eps[theano.config.floatX] assert diff <= type_eps[theano.config.floatX]
@raises(theano.gof.fg.MissingInputError)
def test_strict_mode_ex(self):
n = 10
w = numpy.array([[-1,2],[3,-4]]).astype(theano.config.floatX)
w_ = theano.shared(w)
x0 = numpy.array([1,2]).astype(theano.config.floatX)
x0_ = tensor.vector(name='x0', dtype=theano.config.floatX)
def _scan_loose(x):
return tensor.dot(x, w_)
ret_strict = theano.scan(_scan_loose,
sequences=[],
outputs_info=[x0_],
n_steps=n,
strict=True)
f_strict = theano.function([x0_], ret_strict[0][-1])
result_strict = f_strict(x0)
def test_speed(): def test_speed():
# #
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论