提交 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):
else:
code = """
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)
%(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论