提交 c66e1cf0 authored 作者: Frederic's avatar Frederic

[BUG] with numpy 1.9 numpy.diagonal return a view

上级 93cc458d
......@@ -5119,6 +5119,12 @@ def all(x, axis=None, keepdims=False):
return out
# Some NumPy version like 1.9.2 return a view for numpy.diagonal
x = numpy.zeros((4, 4))
numpy_diagonal_return_view = numpy.may_share_memory(numpy.diagonal(x), x)
del x
class Diagonal(Op):
"""Return specified diagonals.
......@@ -5128,6 +5134,8 @@ class Diagonal(Op):
"""
def __init__(self, offset=0, axis1=0, axis2=1):
if numpy_diagonal_return_view:
self.view_map = {0: [0]}
self.offset = offset
self.axis1 = axis1
self.axis2 = axis2
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论