wehub-resource-sync
91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
# 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__)
|