From 883ccdd5ba6ee4e041630b950c55b3bbd2ed937b Mon Sep 17 00:00:00 2001
From: LoneRanger <836253168@qq.com>
Date: Wed, 2 Aug 2023 15:11:19 +0800
Subject: [PATCH] add the float16 and bfloat16 for poisson (#55824)

---
 python/paddle/tensor/random.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/tensor/random.py b/python/paddle/tensor/random.py
index eb3ab19abc6..b1c2d32c19b 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`.
-- 
GitLab