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

Moved the test_dot_vm and test_dot_mv funcs into TestGemv

上级 49587bc3
...@@ -758,7 +758,8 @@ def test_dot_w_self(): ...@@ -758,7 +758,8 @@ def test_dot_w_self():
f(numpy.asarray([[0,1], [2,3]], dtype=config.floatX)) f(numpy.asarray([[0,1], [2,3]], dtype=config.floatX))
def test_dot_vm(): class TestGemv(TestCase):
def test_dot_vm(self):
''' Test vector dot matrix ''' ''' Test vector dot matrix '''
rng = numpy.random.RandomState(unittest_tools.fetch_seed()) rng = numpy.random.RandomState(unittest_tools.fetch_seed())
v = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32')) v = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32'))
...@@ -774,7 +775,7 @@ def test_dot_vm(): ...@@ -774,7 +775,7 @@ def test_dot_vm():
assert sum([isinstance(node.op, T.blas.Dot22) for node in assert sum([isinstance(node.op, T.blas.Dot22) for node in
f.maker.env.toposort() ]) == 1 f.maker.env.toposort() ]) == 1
def test_dot_mv(): def test_dot_mv(self):
''' Test matrix dot vector ''' ''' Test matrix dot vector '''
rng = numpy.random.RandomState(unittest_tools.fetch_seed()) rng = numpy.random.RandomState(unittest_tools.fetch_seed())
v = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32')) v = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32'))
...@@ -791,7 +792,6 @@ def test_dot_mv(): ...@@ -791,7 +792,6 @@ def test_dot_mv():
assert sum([isinstance(node.op, T.blas.Dot22) for node in assert sum([isinstance(node.op, T.blas.Dot22) for node in
f.maker.env.toposort() ]) == 1 f.maker.env.toposort() ]) == 1
class TestGemv(TestCase):
def test_gemv1(self): def test_gemv1(self):
''' test vector1+dot(matrix,vector2) ''' ''' test vector1+dot(matrix,vector2) '''
rng = numpy.random.RandomState(unittest_tools.fetch_seed()) rng = numpy.random.RandomState(unittest_tools.fetch_seed())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论