[参数完全一致]torch.autograd.Function.forward

torch.autograd.Function.forward

torch.autograd.Function.forward(ctx, *args, **kwargs)

paddle.autograd.PyLayer.forward

paddle.autograd.PyLayer.forward(ctx, *args, **kwargs)

两者功能一致,参数完全一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
ctx ctx 上下文对象。
args args 自定义算子的输入。
kwargs kwargs 自定义算子的输入。