提交 6337d888 authored 作者: Frederic's avatar Frederic

reuse object.

上级 860b3c3c
...@@ -6169,7 +6169,7 @@ class AdvancedSubtensor(Op): ...@@ -6169,7 +6169,7 @@ class AdvancedSubtensor(Op):
gz, = grads gz, = grads
x = inputs[0] x = inputs[0]
rest = inputs[1:] rest = inputs[1:]
return [AdvancedIncSubtensor()(zeros_like(x), gz, return [advanced_inc_subtensor(zeros_like(x), gz,
*rest)] + [None] * len(rest) *rest)] + [None] * len(rest)
...@@ -6262,7 +6262,7 @@ class AdvancedIncSubtensor(Op): ...@@ -6262,7 +6262,7 @@ class AdvancedIncSubtensor(Op):
return [None] return [None]
return self.make_node(eval_points[0], eval_points[1], return self.make_node(eval_points[0], eval_points[1],
*inputs[2:]).outputs *inputs[2:]).outputs
advanced_inc_subtensor = AdvancedIncSubtensor()
######################### #########################
# Linalg : Dot # Linalg : Dot
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论