megengine.autodiff.Function.forward

Function.forward(*args, **kwargs)

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

参数

input – 输入张量。

返回

一个 Tensor 的元组或者单个 Tensor

注解

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

• positional arguments should all be Tensor

上一页

megengine.autodiff.Function.backward

下一页

megengine.autodiff.GradManager