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

fixed test

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