From 36da986385c50ae0a97b9fd47a8df8c8a301a339 Mon Sep 17 00:00:00 2001 From: yangjianfengo1 <125249383+yangjianfengo1@users.noreply.github.com> Date: Tue, 31 Oct 2023 10:36:36 +0800 Subject: [PATCH 1/2] Create top_p_sampling_cn.rst --- docs/api/paddle/top_p_sampling_cn.rst | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 docs/api/paddle/top_p_sampling_cn.rst diff --git a/docs/api/paddle/top_p_sampling_cn.rst b/docs/api/paddle/top_p_sampling_cn.rst new file mode 100644 index 00000000000..2b24947ad66 --- /dev/null +++ b/docs/api/paddle/top_p_sampling_cn.rst @@ -0,0 +1,25 @@ +.. _cn_api_paddle_top_p_sampling: + +top_p_sampling +------------------------------- + +.. py:function:: paddle.top_p_sampling(x, ps, threshold=None, seed=None, name=None) + +从累计概率超过某一个阈值 ``ps``的词汇中进行采样 + +参数 +::::::::: + - **x** (Tensor) - 输入的多维 ``Tensor`` ,支持的数据类型:float32、float16、bfloat16。 + - **ps** (Tensor) - 输入的一维 ``Tensor`` ,长度等于 ``x.shape[0]``,支持的数据类型:float32、float16、bfloat16。 + - **seed** (int,可选) - 指定随机数种子,默认值为-1。 + - **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。 + +返回 +::::::::: +tuple(Tensor), 返回 top_p_sampling 的结果和结果的索引信息。结果的数据类型和输入 ``x`` 一致。索引的数据类型是 int64。 + + +代码示例 +::::::::: + +COPY-FROM: paddle.top_p_sampling From feac3a9451face4990d2b8360bceba5e89c95d54 Mon Sep 17 00:00:00 2001 From: yangjianfengo1 <125249383+yangjianfengo1@users.noreply.github.com> Date: Tue, 31 Oct 2023 10:38:36 +0800 Subject: [PATCH 2/2] Update top_p_sampling_cn.rst --- docs/api/paddle/top_p_sampling_cn.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/api/paddle/top_p_sampling_cn.rst b/docs/api/paddle/top_p_sampling_cn.rst index 2b24947ad66..2648e603e2c 100644 --- a/docs/api/paddle/top_p_sampling_cn.rst +++ b/docs/api/paddle/top_p_sampling_cn.rst @@ -5,7 +5,7 @@ top_p_sampling .. py:function:: paddle.top_p_sampling(x, ps, threshold=None, seed=None, name=None) -从累计概率超过某一个阈值 ``ps``的词汇中进行采样 +从累计概率超过某一个阈值 ``ps`` 的词汇中进行采样 参数 :::::::::