megengine.core.autodiff.grad.Function.forward

Function.forward(*args, **kwargs)[源代码]

对 inputs 应用操作并返回结果。此方法应该被所有派生类重写。

参数

input – 输入张量。

返回

一个 Tensor 的元组或者单个 Tensor

注解

  • 此方法应该返回一个 Tensor 的元组或者单个 Tensor，表示函数的输出。

  • positional arguments should all be Tensor

上一页

megengine.core.autodiff.grad.Function.backward

下一页

megengine.core.autodiff.grad.get_grad_managers