megengine.autodiff.Function.backward

Function.backward(*output_grads)[源代码]

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

参数

output_grads – forward 的输出的梯度。

注解

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

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

上一页

megengine.autodiff.Function.scope

下一页

megengine.autodiff.Function.forward