提交 0f03bac6 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

BatchedDot: PyArray_Newshape complains about NPY_KEEPORDER when it shouldn't

上级 9a3f6681
...@@ -3507,7 +3507,7 @@ class BatchedDot(Op): ...@@ -3507,7 +3507,7 @@ class BatchedDot(Op):
return """{ return """{
npy_intp dims[3] = {%(_shape)s}; npy_intp dims[3] = {%(_shape)s};
PyArray_Dims newshape = {.ptr = dims, .len = 3}; PyArray_Dims newshape = {.ptr = dims, .len = 3};
%(newname)s = (PyArrayObject*)PyArray_Newshape(%(oldname)s, &newshape, NPY_KEEPORDER); %(newname)s = (PyArrayObject*)PyArray_Newshape(%(oldname)s, &newshape, NPY_ANYORDER);
if (!%(newname)s) if (!%(newname)s)
%(_fail)s %(_fail)s
// make sure we didn't accidentally copy // make sure we didn't accidentally copy
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论