diff options
Diffstat (limited to 'tensorflow/contrib/distributions/python/ops/poisson.py')
-rw-r--r-- | tensorflow/contrib/distributions/python/ops/poisson.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/distributions/python/ops/poisson.py b/tensorflow/contrib/distributions/python/ops/poisson.py index 799796ace0..e1ddc9a0e1 100644 --- a/tensorflow/contrib/distributions/python/ops/poisson.py +++ b/tensorflow/contrib/distributions/python/ops/poisson.py @@ -28,7 +28,7 @@ from tensorflow.python.ops import array_ops from tensorflow.python.ops import check_ops from tensorflow.python.ops import control_flow_ops from tensorflow.python.ops import math_ops - +from tensorflow.python.ops import random_ops __all__ = [ "Poisson", @@ -148,6 +148,10 @@ class Poisson(distribution.Distribution): def _mode(self): return math_ops.floor(self.rate) + def _sample_n(self, n, seed=None): + return random_ops.random_poisson( + self.rate, [n], dtype=self.dtype, seed=seed) + def _assert_valid_sample(self, x, check_integer=True): if not self.validate_args: return x |