megengine.core.autodiff.grad.Function.backward

Function.backward(*output_grads)

计算前向函数的梯度。此方法需要被所有派生类重写。

参数

output_grads – forward 的输出的梯度。

注解

• 如果一些输出 Tensor 不影响 loss 函数，那么 output_grads 中的对应值会是 None。

• 此方法应该返回一个包含所有输入的梯度的元组，其顺序应该和 forward 的 inputs 参数一致。如果只有一个输入，那么返回值也可以是单个 Tensor。如果用户希望组织某些梯度的传播，相应的返回值可以被设为 None。

上一页

megengine.core.autodiff.grad.Function.scope

下一页

megengine.core.autodiff.grad.Function.forward