diff options
| author |  Eugene Brevdo <ebrevdo@google.com> | 2017-03-06 09:42:03 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-06 10:08:22 -0800 |
| commit | 988a443383a56619aeaa73354146579addfb13ad (patch) | |
| tree | f3d9baccca474ef8142860b00a242d2f39233374 /tensorflow/contrib/distributions/__init__.py | |
| parent | 540e8d8a7e2c78c2aea366bf6e8af2ade8b74b2d (diff) | |
Expose bijectors as part of the distributions API
For some reason they were hidden in a refactoring to bijector impl.
Change: 149309294
Diffstat (limited to 'tensorflow/contrib/distributions/__init__.py')
| -rw-r--r-- | tensorflow/contrib/distributions/__init__.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/tensorflow/contrib/distributions/__init__.py b/tensorflow/contrib/distributions/__init__.py index b9620a72af..c575ba97a5 100644 --- a/tensorflow/contrib/distributions/__init__.py +++ b/tensorflow/contrib/distributions/__init__.py @@ -16,9 +16,11 @@ See the @{$python/contrib.distributions} guide. +## Distribution Object @@ReparameterizationType @@Distribution +## Individual Distributions @@Binomial @@Bernoulli @@BernoulliWithSigmoidProbs @@ -56,25 +58,26 @@ See the @{$python/contrib.distributions} guide. @@WishartCholesky @@WishartFull -@@matrix_diag_transform - @@TransformedDistribution @@QuantizedDistribution @@Mixture -@@normal_conjugates_known_scale_posterior -@@normal_conjugates_known_scale_predictive +@@ExpRelaxedOneHotCategorical +@@OneHotCategorical +@@RelaxedBernoulli +@@RelaxedOneHotCategorical +## Kullback-Leibler Divergence @@kl @@RegisterKL +## Helper Functions +@@matrix_diag_transform +@@normal_conjugates_known_scale_posterior +@@normal_conjugates_known_scale_predictive @@softplus_inverse -@@ExpRelaxedOneHotCategorical -@@OneHotCategorical -@@RelaxedBernoulli -@@RelaxedOneHotCategorical """ from __future__ import absolute_import from __future__ import division @@ -82,8 +85,10 @@ from __future__ import print_function # pylint: disable=unused-import,wildcard-import,line-too-long,g-importing-member + from tensorflow.contrib.distributions.python.ops.bernoulli import * from tensorflow.contrib.distributions.python.ops.beta import * +from tensorflow.contrib.distributions.python.ops.bijectors import * from tensorflow.contrib.distributions.python.ops.binomial import * from tensorflow.contrib.distributions.python.ops.categorical import * from tensorflow.contrib.distributions.python.ops.chi2 import * |