PyTorch 1.8 与 Paddle 2.0 API映射表

本文档基于X2Paddle研发过程梳理了PyTorch（v1.8.1)常用API与PaddlePaddle 2.0.0 API对应关系与差异分析。通过本文档，帮助开发者快速迁移PyTorch使用经验，完成模型的开发与调优。

X2Paddle介绍

X2Paddle致力于帮助其它主流深度学习框架开发者快速迁移至飞桨框架，目前提供三大功能

  • 预测模型转换

    • 支持Caffe/TensorFlow/ONNX/PyTorch的模型一键转为飞桨的预测模型，并使用PaddleInference/PaddleLite进行CPU/GPU/Arm等设备的部署

  • PyTorch训练项目转换

    • 支持PyTorch项目Python代码（包括训练、预测）一键转为基于飞桨框架的项目代码，帮助开发者快速迁移项目，并可享受AIStudio平台对于飞桨框架提供的海量免费计算资源

  • API映射文档

    • 详细的API文档对比分析，帮助开发者快速从PyTorch框架的使用迁移至飞桨框架的使用，大大降低学习成本

详细的项目信息与使用方法参考X2Paddle在Github上的开源项目: https://github.com/PaddlePaddle/X2Paddle

API映射表目录

类别 简介
基础操作类 主要为torch.XX类API
组网类 主要为torch.nn.XX类下组网相关的API
Loss类 主要为torch.nn.XX类下loss相关的API
工具类 主要为torch.nn.XX类下分布式相关的API和torch.utils.XX类API
视觉类 主要为torchvision.XX类API

基础操作类API映射列表

梳理了基础操作的PyTorch-PaddlePaddle API映射列表，主要包括了构造Tensor、数学计算、逻辑计算相关的API。

序号 PyTorch API PaddlePaddle API 备注
1 torch.set_default_dtype paddle.set_default_dtype 功能一致
2 torch.get_default_dtype paddle.get_default_dtype 功能一致
3 torch.numel paddle.numel 功能一致，参数名不一致
4 torch.tensor paddle.to_tensor 差异对比
5 torch.from_numpy paddle.to_tensor 差异对比
6 torch.zeros paddle.zeros 差异对比
7 torch.zeros_like paddle.zeros_like 差异对比
8 torch.ones paddle.ones 差异对比
9 torch.ones_like paddle.ones_like 差异对比
10 torch.empty paddle.empty 差异对比
11 torch.empty_like paddle.empty_like 差异对比
12 torch.full paddle.full 功能一致，参数不一致
13 torch.full_like paddle.full_like 差异对比
14 torch.arange paddle.arange 功能一致，参数不一致
15 torch.range paddle.arange 功能一致，参数不一致
16 torch.linspace paddle.linspace 功能一致，参数不一致
17 torch.eye paddle.eye 功能一致，参数不一致
18 torch.cat paddle.concat 功能一致，参数名不一致
19 torch.chunk paddle.chunk 功能一致，参数名不一致
20 torch.gather paddle.gather 差异对比
21 torch.index_select paddle.index_select 功能一致，参数名不一致
22 torch.masked_select paddle.masked_select 功能一致，参数名不一致
23 torch.narrow paddle.slice 差异对比
24 torch.nonzero paddle.nonzero 功能一致，参数名不一致
25 torch.reshape paddle.reshape 功能一致，参数名不一致
26 torch.split paddle.split 功能一致，参数名不一致
27 torch.squeeze paddle.squeeze 功能一致，参数名不一致
28 torch.stack paddle.stack 功能一致，参数名不一致
29 torch.t paddle.t 功能一致，参数名不一致
30 torch.transpose paddle.transpose 差异对比
31 torch.unbind paddle.unbind 功能一致，参数名不一致
32 torch.unsqueeze paddle.unsqueeze 功能一致，参数名不一致
33 torch.where paddle.where 功能一致
34 torch.bernoulli paddle.bernoulli 功能一致，参数不一致
35 torch.multinomial paddle.multinomial 功能一致，参数不一致
36 torch.normal paddle.normal 差异对比
37 torch.rand paddle.rand 差异对比
38 torch.randint paddle.randint 功能一致，参数不一致
39 torch.randn paddle.randn 差异对比
40 torch.randperm paddle.randperm 功能一致，参数不一致
41 torch.save paddle.save 差异对比
42 torch.load paddle.load 差异对比
43 torch.abs paddle.abs 功能一致，PaddlePaddle未定义out参数代表输出Tensor
44 torch.absolute paddle.abs 功能一致，PaddlePaddle未定义out参数代表输出Tensor
45 torch.acos paddle.acos 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
46 torch.arccos paddle.acos 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
47 torch.add padle.add 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
48 torch.asin paddle.asin 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
49 torch.arcsin paddle.asin 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
50 torch.atan paddle.atan 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
51 torch.arctan paddle.atan 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
52 torch.ceil paddle.ceil 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
53 torch.clamp paddle.clip 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
54 torch.conj paddle.conj 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
55 torch.cos paddle.cos 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
56 torch.cosh paddle.cosh 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
57 torch.div paddle.divide 差异对比
58 torch.divide paddle.divide 差异对比
59 torch.erf paddle.erf 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
60 torch.exp paddle.exp 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
61 torch.floor paddle.floor 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
62 torch.floor_divide paddle.floor_divide 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
63 torch.fmod paddle.mod 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
64 torch.log paddle.log 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
65 torch.log10 paddle.log10 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
66 torch.log1p paddle.log1p 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
67 torch.log2 paddle.log2 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
68 torch.logical_and paddle.logical_and 功能一致，参数名不一致
69 torch.logical_not paddle.logical_not 功能一致，参数名不一致
70 torch.logical_or paddle.logical_or 功能一致，参数名不一致
71 torch.logical_xor paddle.logical_xor 功能一致，参数名不一致
72 torch.mul paddle.multiply 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
73 torch.multiply paddle.multiply 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
74 torch.pow paddle.pow 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
75 torch.real paddle.real 功能一致，参数名不一致
76 torch.reciprocal paddle.reciprocal 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
77 torch.remainder paddle.mod 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
78 torch.round paddle.round 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
79 torch.rsqrt paddle.rsqrt 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
80 torch.sign paddle.sign 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
81 torch.sin paddle.sin 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
82 torch.sinh paddle.sinh 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
83 torch.sqrt paddle.sqrt 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
84 torch.argmax paddle.argmax 功能一致，参数名不一致
85 torch.argmin paddle.argmin 功能一致，参数名不一致
86 torch.max paddle.max 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor
87 torch.min paddle.min 功能一致，参数名不一致，PaddlePaddle未定义out参数代表输出Tensor

持续更新...

组网类API映射列表

梳理了与构造网络相关的PyTorch-PaddlePaddle API映射列表。

序号 PyTorch API PaddlePaddle API 备注
1 torch.nn.Conv1d paddle.nn.Conv1D 差异对比
2 torch.nn.Conv2d paddle.nn.Conv2D 差异对比
3 torch.nn.Conv3d paddle.nn.Conv3D 差异对比
4 torch.nn.ConvTranspose1d paddle.nn.Conv1DTranspose 差异对比
5 torch.nn.ConvTranspose2d paddle.nn.Conv2DTranspose 差异对比
6 torch.nn.ConvTranspose3d paddle.nn.Conv3DTranspose 差异对比
7 torch.nn.Linear paddle.nn.Linear 差异对比
8 torch.nn.MaxPool1d paddle.nn.MaxPool1D 差异对比
9 torch.nn.MaxPool2d paddle.nn.MaxPool2D 差异对比
10 torch.nn.MaxPool3d paddle.nn.MaxPool3D 差异对比
11 torch.nn.MaxUnpool1d 无对应实现 组合实现
12 torch.nn.MaxUnpool2d 无对应实现 组合实现
13 torch.nn.MaxUnpool3d 无对应实现 组合实现
14 torch.nn.AvgPool1d paddle.nn.AvgPool1D 差异对比
15 torch.nn.AvgPool2d paddle.nn.AvgPool2D 差异对比
16 torch.nn.AvgPool3d paddle.nn.AvgPool3D 差异对比
17 torch.nn.AdaptiveMaxPool1d paddle.nn.AdaptiveMaxPool1D 功能一致，参数名不一致
18 torch.nn.AdaptiveMaxPool2d paddle.nn.AdaptiveMaxPool2D 功能一致，参数名不一致
19 torch.nn.AdaptiveMaxPool3d paddle.nn.AdaptiveMaxPool3D 功能一致，参数名不一致
20 torch.nn.AdaptiveAvgPool1d paddle.nn.AdaptiveAvgPool1D 功能一致，参数名不一致
21 torch.nn.AdaptiveAvgPool2d paddle.nn.AdaptiveAvgPool2D 功能一致，参数名不一致
22 torch.nn.AdaptiveAvgPool3d paddle.nn.AdaptiveAvgPool3D 功能一致，参数名不一致
23 torch.nn.ConstantPad1d paddle.nn.Pad1D 差异对比
24 torch.nn.ConstantPad2d paddle.nn.Pad2D 差异对比
25 torch.nn.ConstantPad3d paddle.nn.Pad3D 差异对比
26 torch.nn.ReflectionPad1d paddle.nn.Pad1D 差异对比
27 torch.nn.ReflectionPad2d paddle.nn.Pad2D 差异对比
28 torch.nn.ReplicationPad1d paddle.nn.Pad1D 差异对比
29 torch.nn.ReplicationPad2d paddle.nn.Pad2D 差异对比
30 torch.nn.ReplicationPad3d paddle.nn.Pad3D 差异对比
31 torch.nn.BatchNorm1d paddle.nn.BatchNorm1D 差异对比
32 torch.nn.BatchNorm2d paddle.nn.BatchNorm2D 差异对比
33 torch.nn.BatchNorm3d paddle.nn.BatchNorm3D 差异对比
34 torch.nn.Upsample paddle.nn.Upsample 差异对比
35 torch.nn.Dropout paddle.nn.Dropout 差异对比
36 torch.nn.Dropout2d paddle.nn.Dropout2D 差异对比
37 torch.nn.Dropout3d paddle.nn.Dropout3D 差异对比
38 torch.nn.LSTM paddle.nn.LSTM 差异对比
39 torch.nn.GRU paddle.nn.GRU 差异对比
40 torch.nn.Embedding paddle.nn.Embedding 差异对比
41 torch.nn.ELU paddle.nn.ELU 功能一致，PaddlePaddle未定义inplace参数表示在不更改变量的内存地址的情况下，直接修改变量的值
42 torch.nn.Hardsigmoid paddle.nn.Hardsigmoid 功能一致，PaddlePaddle未定义inplace参数表示在不更改变量的内存地址的情况下，直接修改变量的值
43 torch.nn.LeakyReLU paddle.nn.LeakyReLU 功能一致，PaddlePaddle未定义inplace参数表示在不更改变量的内存地址的情况下，直接修改变量的值
44 torch.nn.PReLU paddle.nn.PReLU 功能一致
45 torch.nn.ReLU paddle.nn.ReLU 功能一致，PaddlePaddle未定义inplace参数表示在不更改变量的内存地址的情况下，直接修改变量的值
46 torch.nn.Softmax paddle.nn.Softmax 功能一致，参数名不一致

持续更新...

Loss类API映射列表

梳理了计算loss相关的PyTorch-PaddlePaddle API映射列表。

序号 PyTorch API PaddlePaddle API 备注
1 torch.nn.L1Loss paddle.nn.L1Loss 功能一致，PyTroch存在废弃参数size_average和reduce。
2 torch.nn.MSELoss paddle.nn.MSELoss 功能一致，PyTroch存在废弃参数size_average和reduce。
3 torch.nn.CrossEntropyLoss paddle.nn.CrossEntropyLoss 差异对比
4 torch.nn.KLDivLoss paddle.nn.KLDivLoss 差异对比
5 torch.nn.BCELoss paddle.nn.BCELoss 功能一致，PyTroch存在废弃参数size_average和reduce。
6 torch.nn.BCEWithLogitsLoss paddle.nn.BCEWithLogitsLoss 功能一致，PyTroch存在废弃参数size_average和reduce。
7 torch.nn.SmoothL1Loss paddle.nn.SmoothL1Loss 功能一致，参数名不一致，PyTroch存在废弃参数size_average和reduce。

持续更新...

视觉类API映射列表

梳理了与视觉处理相关的PyTorch-PaddlePaddle API映射列表。

序号 PyTorch API PaddlePaddle API 备注
1 torchvision.transforms.Compose paddle.vision.transforms.Compose 功能一致
2 torchvision.transforms.ToPILImage 无对应实现 组合实现
3 torchvision.transforms.Resize paddle.vision.transforms.Resize 功能一致
4 torchvision.transforms.ToTensor paddle.vision.transforms.ToTensor 功能一致
5 torchvision.transforms.RandomHorizontalFlip paddle.vision.transforms.RandomHorizontalFlip 功能一致
6 torchvision.transforms.CenterCrop paddle.vision.transforms.CenterCrop 功能一致
7 torchvision.transforms.ColorJitter paddle.vision.transforms.ColorJitter 功能一致
8 torchvision.transforms.Grayscale paddle.vision.transforms.Grayscale 功能一致
9 torchvision.transforms.Normalize paddle.vision.transforms.Normalize 差异对比
10 torchvision.transforms.RandomResizedCrop paddle.vision.transforms.RandomResizedCrop 功能一致
11 torchvision.transforms.Pad paddle.vision.transforms.Pad 功能一致
12 torchvision.transforms.RandomCrop paddle.vision.transforms.RandomCrop 功能一致
13 torchvision.transforms.RandomRotation paddle.vision.transforms.RandomRotation 功能一致
14 torchvision.transforms.RandomVerticalFlip paddle.vision.transforms.RandomVerticalFlip 功能一致
15 torchvision.transforms.Lambda 无对应实现 组合实现
17 torchvision.utils.save_image 无对应实现 组合实现
18 torchvision.models 系列模型 X2Paddle提供 使用方式

持续更新...