chore: import upstream snapshot with attribution

python/paddle/distribution/__init__.py

@@ -0,0 +1,90 @@
+# Copyright (c) 2021 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.
+
+from . import constraint as constraint, transform
+from .bernoulli import Bernoulli
+from .beta import Beta
+from .binomial import Binomial
+from .categorical import Categorical
+from .cauchy import Cauchy
+from .chi2 import Chi2
+from .continuous_bernoulli import ContinuousBernoulli
+from .dirichlet import Dirichlet
+from .distribution import Distribution
+from .exponential import Exponential
+from .exponential_family import ExponentialFamily
+from .gamma import Gamma
+from .geometric import Geometric
+from .gumbel import Gumbel
+from .independent import Independent
+from .kl import kl_divergence, register_kl
+from .laplace import Laplace
+from .lkj_cholesky import LKJCholesky
+from .lognormal import LogNormal
+from .multinomial import Multinomial
+from .multivariate_normal import MultivariateNormal
+from .normal import Normal
+from .poisson import Poisson
+from .student_t import StudentT
+from .transform import (  # noqa:F401
+    AbsTransform,
+    AffineTransform,
+    ChainTransform,
+    ExpTransform,
+    IndependentTransform,
+    PowerTransform,
+    ReshapeTransform,
+    SigmoidTransform,
+    SoftmaxTransform,
+    StackTransform,
+    StickBreakingTransform,
+    TanhTransform,
+    Transform,
+)
+from .transformed_distribution import TransformedDistribution
+from .uniform import Uniform
+
+constraints = constraint
+
+__all__ = [
+    'Bernoulli',
+    'Beta',
+    'Categorical',
+    'Cauchy',
+    'Chi2',
+    'ContinuousBernoulli',
+    'Dirichlet',
+    'Distribution',
+    'Exponential',
+    'ExponentialFamily',
+    'Multinomial',
+    'MultivariateNormal',
+    'Normal',
+    'Uniform',
+    'kl_divergence',
+    'register_kl',
+    'Independent',
+    'TransformedDistribution',
+    'Laplace',
+    'LogNormal',
+    'LKJCholesky',
+    'Gamma',
+    'Gumbel',
+    'Geometric',
+    'Binomial',
+    'Poisson',
+    'StudentT',
+]
+
+__all__.extend(transform.__all__)