diff options
Diffstat (limited to 'tensorflow/python/ops/distributions/dirichlet.py')
-rw-r--r-- | tensorflow/python/ops/distributions/dirichlet.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/ops/distributions/dirichlet.py b/tensorflow/python/ops/distributions/dirichlet.py index 923696a553..2accedf1b9 100644 --- a/tensorflow/python/ops/distributions/dirichlet.py +++ b/tensorflow/python/ops/distributions/dirichlet.py @@ -196,7 +196,7 @@ class Dirichlet(distribution.Distribution): alpha=self.concentration, dtype=self.dtype, seed=seed) - return gamma_sample / math_ops.reduce_sum(gamma_sample, -1, keep_dims=True) + return gamma_sample / math_ops.reduce_sum(gamma_sample, -1, keepdims=True) @distribution_util.AppendDocstring(_dirichlet_sample_note) def _log_prob(self, x): |