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