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

fixed bug

上级 164d2c78
...@@ -117,7 +117,7 @@ class MultinomialFromUniform(Op): ...@@ -117,7 +117,7 @@ class MultinomialFromUniform(Op):
{ {
int waiting = 1; int waiting = 1;
dtype_%(pvals)s cummul = 0.; dtype_%(pvals)s cummul = 0.;
const dtype_%(unis)s* unis_n = (dtype_%(unis)s*)PyArray_GETPTR1(%(unis)s, c*n_samples + n); const dtype_%(unis)s* unis_n = (dtype_%(unis)s*)PyArray_GETPTR1(%(unis)s, c*nb_multi + n);
for (int m = 0; m < nb_outcomes; ++m) for (int m = 0; m < nb_outcomes; ++m)
{ {
dtype_%(z)s* z_nm = (dtype_%(z)s*)PyArray_GETPTR2(%(z)s, n,m); dtype_%(z)s* z_nm = (dtype_%(z)s*)PyArray_GETPTR2(%(z)s, n,m);
...@@ -150,8 +150,6 @@ class MultinomialFromUniform(Op): ...@@ -150,8 +150,6 @@ class MultinomialFromUniform(Op):
""" % locals() """ % locals()
def perform(self, node, ins, outs): def perform(self, node, ins, outs):
# import pdb; pdb.set_trace()
(pvals, unis, n_samples) = ins (pvals, unis, n_samples) = ins
(z,) = outs (z,) = outs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论