From 5e886207e2c6a7674698d38297a5d68c04fcaf84 Mon Sep 17 00:00:00 2001
From: longranger2 <836253168@qq.com>
Date: Mon, 31 Jul 2023 11:43:45 +0800
Subject: [PATCH] add the float16 and bfloat16 for poisson

---
 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 eb3ab19abc6ed..b1c2d32c19b19 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`.