diff --git a/tester/paddle_to_torch/mapping.json b/tester/paddle_to_torch/mapping.json index 96a174b7..01f40452 100644 --- a/tester/paddle_to_torch/mapping.json +++ b/tester/paddle_to_torch/mapping.json @@ -3100,9 +3100,7 @@ }, "paddle.Tensor.median": { "torch_api": "torch.Tensor.median", - "paddle_torch_args_map": { - "keepdim": "keepdim" - } + "Rule": "MedianRule" }, "paddle.Tensor.mm": { "torch_api": "torch.Tensor.mm",