提交 b4cc877f authored 作者: Frederic Bastien's avatar Frederic Bastien

[BUG] fix a bug recently introduced. It was detected by the DailyBuildbot in DebugMode.

上级 be93a84e
...@@ -6105,6 +6105,8 @@ class ExtractDiag(Op): ...@@ -6105,6 +6105,8 @@ class ExtractDiag(Op):
(x,) = inputs (x,) = inputs
(z,) = outputs (z,) = outputs
z[0] = x.diagonal(self.offset, self.axis1, self.axis2) z[0] = x.diagonal(self.offset, self.axis1, self.axis2)
if not self.view:
z[0] = z[0].copy()
def grad(self, inputs, gout): def grad(self, inputs, gout):
(x,) = inputs (x,) = inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论