Skip to content
Merged
37 changes: 37 additions & 0 deletions python/paddle/nn/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,9 @@
from .layer.common import (
AlphaDropout,
Bilinear,
ConstantPad1D,
ConstantPad2D,
ConstantPad3D,
CosineSimilarity,
Dropout,
Dropout2D,
Expand All @@ -73,6 +76,12 @@
Pad1D,
Pad2D,
Pad3D,
ReflectionPad1D,
ReflectionPad2D,
ReflectionPad3D,
ReplicationPad1D,
ReplicationPad2D,
ReplicationPad3D,
Unflatten,
Unfold,
Upsample,
Expand Down Expand Up @@ -189,6 +198,16 @@
from .utils.spectral_norm_hook import spectral_norm # noqa: F401

SiLU = Silu
ReflectionPad1d = ReflectionPad1D
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

你也可以conv2d=Conv2D那些也挪出来,统一放一个位置。

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done

ReflectionPad2d = ReflectionPad2D
ReflectionPad3d = ReflectionPad3D
ConstantPad1d = ConstantPad1D
ConstantPad2d = ConstantPad2D
ConstantPad3d = ConstantPad3D
ReplicationPad1d = ReplicationPad1D
ReplicationPad2d = ReplicationPad2D
ReplicationPad3d = ReplicationPad3D


__all__ = [
'BatchNorm',
Expand All @@ -212,6 +231,24 @@
'Pad1D',
'Pad2D',
'Pad3D',
'ConstantPad1D',
'ConstantPad2D',
'ConstantPad3D',
'ReplicationPad1D',
'ReplicationPad2D',
'ReplicationPad3D',
'ReflectionPad1D',
'ReflectionPad2D',
'ReflectionPad3D',
'ConstantPad1d',
'ConstantPad2d',
'ConstantPad3d',
'ReplicationPad1d',
'ReplicationPad2d',
'ReplicationPad3d',
'ReflectionPad1d',
'ReflectionPad2d',
'ReflectionPad3d',
'CosineSimilarity',
'Dropout',
'Dropout2D',
Expand Down
9 changes: 9 additions & 0 deletions python/paddle/nn/layer/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,9 @@
from .common import ( # noqa: F401
AlphaDropout,
Bilinear,
ConstantPad1D,
ConstantPad2D,
ConstantPad3D,
CosineSimilarity,
Dropout,
Dropout2D,
Expand All @@ -43,6 +46,12 @@
Pad1D,
Pad2D,
Pad3D,
ReflectionPad1D,
ReflectionPad2D,
ReflectionPad3D,
ReplicationPad1D,
ReplicationPad2D,
ReplicationPad3D,
Unflatten,
Upsample,
UpsamplingBilinear2D,
Expand Down
Loading
Loading