未验证 提交 883ccdd5 编写于 作者: L LoneRanger 提交者: GitHub

add the float16 and bfloat16 for poisson (#55824)

上级 92aa92fa
......@@ -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`.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册