diff --git a/python/paddle/tensor/random.py b/python/paddle/tensor/random.py index eb3ab19abc6edcd1df491f48ea53282cb0d6920f..b1c2d32c19b1924d48b22a882078567efd3acfe7 100644 --- a/python/paddle/tensor/random.py +++ b/python/paddle/tensor/random.py @@ -102,7 +102,7 @@ def poisson(x, name=None): Args: x(Tensor): A tensor with rate parameter of poisson Distribution. The data type - should be float32, float64. + should be bfloat16, float16, float32, float64. name(str, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name`.