提交 2dddf4ac authored 作者: Marc-Alexandre Cote's avatar Marc-Alexandre Cote

Changed PyArray_SHAPE to PyArray_DIMS.

上级 c9d6cfe0
...@@ -74,7 +74,7 @@ class CumsumOp(theano.Op): ...@@ -74,7 +74,7 @@ class CumsumOp(theano.Op):
else: else:
code = """ code = """
Py_XDECREF(%(z)s); Py_XDECREF(%(z)s);
%(z)s = (PyArrayObject*) PyArray_SimpleNew(PyArray_NDIM(%(x)s), PyArray_SHAPE(%(x)s), type_num_%(x)s); %(z)s = (PyArrayObject*) PyArray_SimpleNew(PyArray_NDIM(%(x)s), PyArray_DIMS(%(x)s), type_num_%(x)s);
if (!%(z)s) if (!%(z)s)
%(fail)s; %(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论