Skip to content

Commit 064e8a9

Browse files
committed
add api and test
1 parent a46df40 commit 064e8a9

File tree

7 files changed

+2080
-0
lines changed

7 files changed

+2080
-0
lines changed

python/paddle/distribution/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
from paddle.distribution.beta import Beta
1818
from paddle.distribution.categorical import Categorical
1919
from paddle.distribution.cauchy import Cauchy
20+
from paddle.distribution.continuous_bernoulli import ContinuousBernoulli
2021
from paddle.distribution.dirichlet import Dirichlet
2122
from paddle.distribution.distribution import Distribution
2223
from paddle.distribution.gumbel import Gumbel
@@ -25,6 +26,7 @@
2526
from paddle.distribution.kl import kl_divergence, register_kl
2627
from paddle.distribution.lognormal import LogNormal
2728
from paddle.distribution.multinomial import Multinomial
29+
from paddle.distribution.multivariate_normal import MultivariateNormal
2830
from paddle.distribution.normal import Normal
2931
from paddle.distribution.transform import * # noqa: F403
3032
from paddle.distribution.transformed_distribution import TransformedDistribution
@@ -37,10 +39,12 @@
3739
'Beta',
3840
'Categorical',
3941
'Cauchy',
42+
'ContinuousBernoulli',
4043
'Dirichlet',
4144
'Distribution',
4245
'ExponentialFamily',
4346
'Multinomial',
47+
'MultivariateNormal',
4448
'Normal',
4549
'Uniform',
4650
'kl_divergence',

0 commit comments

Comments
 (0)