提交 11c0d243 authored 作者: Frederic Bastien's avatar Frederic Bastien

code simplification.

上级 e04419cc
......@@ -347,11 +347,11 @@ class CumOp(theano.Op):
if(%(params)s->mode == MODE_ADD)
t = PyArray_CumSum(
%(x)s, axis,
PyArray_TYPE((PyArrayObject*) py_%(x)s), %(z)s);
PyArray_TYPE(%(x)s), %(z)s);
else if(%(params)s->mode == MODE_MUL)
t = PyArray_CumProd(
%(x)s, axis,
PyArray_TYPE((PyArrayObject*) py_%(x)s), %(z)s);
PyArray_TYPE(%(x)s), %(z)s);
if (!t){
%(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论