[参数完全一致]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 | 自定义算子的输入。 |