提交 e1f691c3 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

fixed test

上级 f224de0c
...@@ -32,7 +32,7 @@ def run_with_c(f, gpu=False): ...@@ -32,7 +32,7 @@ def run_with_c(f, gpu=False):
def test_n_samples_1(): def test_n_samples_1():
p = tensor.fmatrix() p = tensor.fmatrix()
u = tensor.fvector() u = tensor.fvector()
n = tensor.scalar() n = tensor.iscalar()
m = multinomial.MultinomialFromUniform('auto')(p, u, n) m = multinomial.MultinomialFromUniform('auto')(p, u, n)
f = function([p, u, n], m, allow_input_downcast=True) f = function([p, u, n], m, allow_input_downcast=True)
...@@ -47,7 +47,7 @@ def test_n_samples_1(): ...@@ -47,7 +47,7 @@ def test_n_samples_1():
def test_n_samples_2(): def test_n_samples_2():
p = tensor.fmatrix() p = tensor.fmatrix()
u = tensor.fvector() u = tensor.fvector()
n = tensor.scalar() n = tensor.iscalar()
m = multinomial.MultinomialFromUniform('auto')(p, u, n) m = multinomial.MultinomialFromUniform('auto')(p, u, n)
f = function([p, u, n], m, allow_input_downcast=True) f = function([p, u, n], m, allow_input_downcast=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论