提交 a0048ab1 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

R operator of AdvancedIncSubtensor

上级 937135df
......@@ -4996,6 +4996,11 @@ class AdvancedIncSubtensor(Op):
# grad on x is grad on output
# grad on y is grad_output[idx_list]
# grad on rest is None
def R_op(self, inputs, eval_points):
if None in eval_points[:2]:
return [None]
return self.make_node(eval_points[0], eval_points[1], *inputs[2:]).outputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论