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
        return dy * y * (1-y)

属性

scope

方法

backward(*output_grads)

Compute the gradient of the forward function.

forward(*args, **kwargs)

Applies operations to inputs and returns results.