提交 6d6dcb46 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

remove unnecessary initialization, flake8 fixes

上级 f2451453
...@@ -13,6 +13,7 @@ if cuda_available: ...@@ -13,6 +13,7 @@ if cuda_available:
from theano.sandbox.cuda.basic_ops import host_from_gpu, gpu_from_host from theano.sandbox.cuda.basic_ops import host_from_gpu, gpu_from_host
from theano.sandbox.cuda.opt import register_opt from theano.sandbox.cuda.opt import register_opt
class MultinomialFromUniform(Op): class MultinomialFromUniform(Op):
""" """
Converts samples from a uniform into sample from a multinomial. Converts samples from a uniform into sample from a multinomial.
...@@ -92,7 +93,7 @@ class MultinomialFromUniform(Op): ...@@ -92,7 +93,7 @@ class MultinomialFromUniform(Op):
) )
{ {
Py_XDECREF(%(z)s); Py_XDECREF(%(z)s);
%(z)s = (PyArrayObject*) PyArray_ZEROS(2, %(z)s = (PyArrayObject*) PyArray_EMPTY(2,
PyArray_DIMS(%(pvals)s), PyArray_DIMS(%(pvals)s),
%(t)s, %(t)s,
0); 0);
...@@ -108,7 +109,7 @@ class MultinomialFromUniform(Op): ...@@ -108,7 +109,7 @@ class MultinomialFromUniform(Op):
const int nb_multi = PyArray_DIMS(%(pvals)s)[0]; const int nb_multi = PyArray_DIMS(%(pvals)s)[0];
const int nb_outcomes = PyArray_DIMS(%(pvals)s)[1]; const int nb_outcomes = PyArray_DIMS(%(pvals)s)[1];
const int n_samples = %(n)s; const int n_samples = %(n)s;
// //
// For each multinomial, loop over each possible outcome // For each multinomial, loop over each possible outcome
// //
...@@ -168,7 +169,7 @@ class MultinomialFromUniform(Op): ...@@ -168,7 +169,7 @@ class MultinomialFromUniform(Op):
waiting = True waiting = True
cummul = 0 cummul = 0
unis_n = unis[n] unis_n = unis[n]
for m in range(nb_outcomes): for m in range(nb_outcomes):
cummul += pvals[n, m] cummul += pvals[n, m]
if c == 0: if c == 0:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论