提交 caa214c7 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add grad2 method.

上级 f66ea7e8
......@@ -696,6 +696,9 @@ class PureOp(object):
# Python implementation #
#########################
def grad2(self, inputs, outputs, output_grads):
return self.grad(inputs, output_grads)
def R_op(self, inputs, eval_points):
"""
This method is primarily used by tensor.Rop
......
......@@ -1102,7 +1102,8 @@ def _populate_grad_dict(var_to_app_to_idx,
str(o_shape) + " on an output of shape " +
str(g_shape))
input_grads = node.op.grad(inputs, new_output_grads)
input_grads = node.op.grad2(inputs, node.outputs,
new_output_grads)
if input_grads is None:
raise TypeError("%s.grad returned NoneType, "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论