Skip to content

Commit 41c476b

Browse files
[API Compatibility] Add nn.Constant/Reflection/Replication/Circular Pad layers (#76136)
* Add nn.Constant/Reflection/Replication Pad layers * Refine on comment: Rename 1d/2d/3d to 1D2D3D * Fix sample code * Fix test error * Rerun CI * Refactor pad layers * Fix zeropad test * Fix unittest * Add CircularPad1D/2D/3D * Add CircularPad1D/2D/3D * Skip circular pad 3d test on xpu * Skip circular pad ut on XPU * correct return type of _npairs * correct return type of _npairs with concrete type Co-authored-by: Nyakku Shigure <sigure.qaq@gmail.com> --------- Co-authored-by: Nyakku Shigure <sigure.qaq@gmail.com>
1 parent cbf25c8 commit 41c476b

11 files changed

+1983
-162
lines changed

python/paddle/nn/__init__.py

Lines changed: 49 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,12 @@
6060
from .layer.common import (
6161
AlphaDropout,
6262
Bilinear,
63+
CircularPad1D,
64+
CircularPad2D,
65+
CircularPad3D,
66+
ConstantPad1D,
67+
ConstantPad2D,
68+
ConstantPad3D,
6369
CosineSimilarity,
6470
Dropout,
6571
Dropout2D,
@@ -73,6 +79,12 @@
7379
Pad1D,
7480
Pad2D,
7581
Pad3D,
82+
ReflectionPad1D,
83+
ReflectionPad2D,
84+
ReflectionPad3D,
85+
ReplicationPad1D,
86+
ReplicationPad2D,
87+
ReplicationPad3D,
7688
Unflatten,
7789
Unfold,
7890
Upsample,
@@ -94,13 +106,10 @@
94106
)
95107
from .layer.conv import (
96108
Conv1D,
97-
Conv1d,
98109
Conv1DTranspose,
99110
Conv2D,
100-
Conv2d,
101111
Conv2DTranspose,
102112
Conv3D,
103-
Conv3d,
104113
Conv3DTranspose,
105114
)
106115
from .layer.distance import PairwiseDistance
@@ -189,6 +198,22 @@
189198
from .utils.spectral_norm_hook import spectral_norm # noqa: F401
190199

191200
SiLU = Silu
201+
ReflectionPad1d = ReflectionPad1D
202+
ReflectionPad2d = ReflectionPad2D
203+
ReflectionPad3d = ReflectionPad3D
204+
ConstantPad1d = ConstantPad1D
205+
ConstantPad2d = ConstantPad2D
206+
ConstantPad3d = ConstantPad3D
207+
ReplicationPad1d = ReplicationPad1D
208+
ReplicationPad2d = ReplicationPad2D
209+
ReplicationPad3d = ReplicationPad3D
210+
CircularPad1d = CircularPad1D
211+
CircularPad2d = CircularPad2D
212+
CircularPad3d = CircularPad3D
213+
Conv1d = Conv1D
214+
Conv2d = Conv2D
215+
Conv3d = Conv3D
216+
192217

193218
__all__ = [
194219
'BatchNorm',
@@ -212,6 +237,27 @@
212237
'Pad1D',
213238
'Pad2D',
214239
'Pad3D',
240+
'ConstantPad1D',
241+
'ConstantPad2D',
242+
'ConstantPad3D',
243+
'CircularPad1d',
244+
'CircularPad2d',
245+
'CircularPad3d',
246+
'ReplicationPad1D',
247+
'ReplicationPad2D',
248+
'ReplicationPad3D',
249+
'ReflectionPad1D',
250+
'ReflectionPad2D',
251+
'ReflectionPad3D',
252+
'ConstantPad1d',
253+
'ConstantPad2d',
254+
'ConstantPad3d',
255+
'ReplicationPad1d',
256+
'ReplicationPad2d',
257+
'ReplicationPad3d',
258+
'ReflectionPad1d',
259+
'ReflectionPad2d',
260+
'ReflectionPad3d',
215261
'CosineSimilarity',
216262
'Dropout',
217263
'Dropout2D',

python/paddle/nn/layer/__init__.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,9 @@
3030
from .common import ( # noqa: F401
3131
AlphaDropout,
3232
Bilinear,
33+
ConstantPad1D,
34+
ConstantPad2D,
35+
ConstantPad3D,
3336
CosineSimilarity,
3437
Dropout,
3538
Dropout2D,
@@ -43,6 +46,12 @@
4346
Pad1D,
4447
Pad2D,
4548
Pad3D,
49+
ReflectionPad1D,
50+
ReflectionPad2D,
51+
ReflectionPad3D,
52+
ReplicationPad1D,
53+
ReplicationPad2D,
54+
ReplicationPad3D,
4655
Unflatten,
4756
Upsample,
4857
UpsamplingBilinear2D,

0 commit comments

Comments
 (0)