提交 6402f735 authored 作者: amrithasuresh's avatar amrithasuresh

Updated numpy as np

上级 31c100b5
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import numpy import numpy as np
import warnings import warnings
import theano import theano
...@@ -172,7 +172,7 @@ class MultinomialFromUniform(Op): ...@@ -172,7 +172,7 @@ class MultinomialFromUniform(Op):
raise ValueError("unis.shape[0] != pvals.shape[0] * n_samples", raise ValueError("unis.shape[0] != pvals.shape[0] * n_samples",
unis.shape[0], pvals.shape[0], n_samples) unis.shape[0], pvals.shape[0], n_samples)
if z[0] is None or z[0].shape != pvals.shape: if z[0] is None or z[0].shape != pvals.shape:
z[0] = numpy.zeros(pvals.shape, dtype=node.outputs[0].dtype) z[0] = np.zeros(pvals.shape, dtype=node.outputs[0].dtype)
else: else:
z[0].fill(0) z[0].fill(0)
...@@ -209,7 +209,7 @@ class MultinomialFromUniform(Op): ...@@ -209,7 +209,7 @@ class MultinomialFromUniform(Op):
# have the same answer as the c code as in the c code # have the same answer as the c code as in the c code
# the cumul is in double precission. # the cumul is in double precission.
cumsum = pvals[n].cumsum(dtype='float64') cumsum = pvals[n].cumsum(dtype='float64')
z[0][n, numpy.searchsorted(cumsum, unis_n)] += 1 z[0][n, np.searchsorted(cumsum, unis_n)] += 1
class ChoiceFromUniform(MultinomialFromUniform): class ChoiceFromUniform(MultinomialFromUniform):
...@@ -380,8 +380,8 @@ class ChoiceFromUniform(MultinomialFromUniform): ...@@ -380,8 +380,8 @@ class ChoiceFromUniform(MultinomialFromUniform):
else: else:
odtype = self.odtype odtype = self.odtype
if (z[0] is None or if (z[0] is None or
not numpy.all(z[0].shape == [pvals.shape[0], n_samples])): not np.all(z[0].shape == [pvals.shape[0], n_samples])):
z[0] = -1 * numpy.ones((pvals.shape[0], n_samples), dtype=odtype) z[0] = -1 * np.ones((pvals.shape[0], n_samples), dtype=odtype)
nb_multi = pvals.shape[0] nb_multi = pvals.shape[0]
nb_outcomes = pvals.shape[1] nb_outcomes = pvals.shape[1]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论