megengine.autodiff.Function

class Function

定义一段具有自定义求导规则的计算

计算在 forward 方法中定义，求导规则在 backward 方法中定义

每个 Function 的实例仅应在前向计算中使用一次

实际案例

class Sigmoid(Function):
    def forward(self, x):
        y = 1 / (1 + F.exp(-x))
        self.y = y
        return y

    def backward(self, dy):
        y = self.y

属性

scope

方法

backward(*output_grads)	Compute the gradient of the forward function.
forward(*args, **kwargs)	Applies operations to inputs and returns results.

上一页

megengine.autodiff

下一页

megengine.autodiff.Function.scope