forked from PaddlePaddle/Paddle
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path__init__.py
More file actions
128 lines (125 loc) · 2.77 KB
/
__init__.py
File metadata and controls
128 lines (125 loc) · 2.77 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# TODO: define activation functions of neural network
from . import container, rnn, transformer # noqa: F401
from .activation import ( # noqa: F401
CELU,
LeakyReLU,
LogSoftmax,
PReLU,
ReLU,
ReLU6,
RReLU,
Sigmoid,
Softmax,
Softmax2D,
)
from .common import ( # noqa: F401
AlphaDropout,
Bilinear,
ConstantPad1D,
ConstantPad2D,
ConstantPad3D,
CosineSimilarity,
Dropout,
Dropout2D,
Dropout3D,
Embedding,
FeatureAlphaDropout,
Flatten,
Fold,
Identity,
Linear,
Pad1D,
Pad2D,
Pad3D,
ReflectionPad1D,
ReflectionPad2D,
ReflectionPad3D,
ReplicationPad1D,
ReplicationPad2D,
ReplicationPad3D,
Unflatten,
Upsample,
UpsamplingBilinear2D,
UpsamplingNearest2D,
ZeroPad2D,
)
from .container import LayerDict # noqa: F401
from .conv import ( # noqa: F401
Conv1D,
Conv1DTranspose,
Conv2D,
Conv2DTranspose,
Conv3D,
Conv3DTranspose,
)
from .distance import PairwiseDistance # noqa: F401
from .layers import Layer # noqa: F401
from .loss import ( # noqa: F401
AdaptiveLogSoftmaxWithLoss,
BCELoss,
BCEWithLogitsLoss,
CrossEntropyLoss,
CTCLoss,
GaussianNLLLoss,
HingeEmbeddingLoss,
KLDivLoss,
L1Loss,
MarginRankingLoss,
MSELoss,
MultiLabelMarginLoss,
MultiLabelSoftMarginLoss,
MultiMarginLoss,
NLLLoss,
PoissonNLLLoss,
RNNTLoss,
SmoothL1Loss,
SoftMarginLoss,
TripletMarginLoss,
TripletMarginWithDistanceLoss,
)
from .norm import ( # noqa: F401
BatchNorm1D,
BatchNorm2D,
BatchNorm3D,
GroupNorm,
LayerNorm,
LocalResponseNorm,
SpectralNorm,
SyncBatchNorm,
)
from .pooling import ( # noqa: F401
AdaptiveAvgPool1D,
AdaptiveAvgPool2D,
AdaptiveAvgPool3D,
AdaptiveMaxPool1D,
AdaptiveMaxPool2D,
AdaptiveMaxPool3D,
AvgPool1D,
AvgPool2D,
AvgPool3D,
FractionalMaxPool2D,
FractionalMaxPool3D,
LPPool1D,
LPPool2D,
MaxPool1D,
MaxPool2D,
MaxPool3D,
MaxUnPool1D,
MaxUnPool2D,
MaxUnPool3D,
)
from .vision import ChannelShuffle, PixelShuffle, PixelUnshuffle # noqa: F401
__all__ = []