提交 67d88d11 authored 作者: Frederic Bastien's avatar Frederic Bastien

make better test.

上级 48bc80fb
...@@ -167,7 +167,7 @@ class T_CrossentropySoftmaxArgmax1HotWithBias(unittest.TestCase): ...@@ -167,7 +167,7 @@ class T_CrossentropySoftmaxArgmax1HotWithBias(unittest.TestCase):
var = self.op(numpy.random.rand(3,5),numpy.random.rand(5), y_idx=numpy.random.randint( var = self.op(numpy.random.rand(3,5),numpy.random.rand(5), y_idx=numpy.random.randint(
low=0, high=5, size=3)) low=0, high=5, size=3))
assert theano.function([],var[0].shape)() == [3] assert theano.function([],var[0].shape)() == [3]
assert any(theano.function([],var[1].shape)() == [3,5]) assert all(theano.function([],var[1].shape)() == [3,5])
assert theano.function([],var[2].shape)() == [3] assert theano.function([],var[2].shape)() == [3]
class T_prepend(unittest.TestCase): class T_prepend(unittest.TestCase):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论