提交 f3ab7ed8 authored 作者: Frederic Bastien's avatar Frederic Bastien

small extra speed up

上级 51b082c6
......@@ -208,7 +208,7 @@ class MultinomialFromUniform(Op):
# have the same answer as the c code as in the c code
# the cumul is in double precission.
cumsum = pvals[n].cumsum(dtype='float64')
z[0][n, (cumsum > unis_n).nonzero()[0][0]] += 1
z[0][n, numpy.searchsorted(cumsum, unis_n)] += 1
class MultinomialWOReplacementFromUniform(MultinomialFromUniform):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论