Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions docs/api/paddle/distributed/stream/alltoall_single_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@ alltoall_single

参数
:::::::::
- **out_tensor** (Tensor): 用于保存操作结果的 tensor,数据类型必须与输入的 tensor 保持一致。
- **in_tensor** (Tensor): 输入的 tensor。支持的数据类型包括:float16、float32、float64、int32、int64、int8、uint8、bool、bfloat16。
- **out_split_sizes** (List[int],可选): 对 out_tensor 的 dim[0] 进行切分的大小。默认为 None,即 out_tensor 将均匀地聚合各个进程的数据(需要确保 out_tensor 的大小能够被组中的进程数整除)。
- **in_split_sizes** (List[int],可选): 对 in_tensor 的 dim[0] 进行切分的大小。默认为 None,即将 in_tensor 均匀地分发到各个进程中(需要确保 in_tensor 的大小能够被组中的进程数整除)。
- **out_tensor** (Tensor) - 用于保存操作结果的 tensor,数据类型必须与输入的 tensor 保持一致。
- **in_tensor** (Tensor) - 输入的 tensor。支持的数据类型包括:float16、float32、float64、int32、int64、int8、uint8、bool、bfloat16。
- **out_split_sizes** (List[int],可选) - 对 out_tensor 的 dim[0] 进行切分的大小。默认为 None,即 out_tensor 将均匀地聚合各个进程的数据(需要确保 out_tensor 的大小能够被组中的进程数整除)。
- **in_split_sizes** (List[int],可选) - 对 in_tensor 的 dim[0] 进行切分的大小。默认为 None,即将 in_tensor 均匀地分发到各个进程中(需要确保 in_tensor 的大小能够被组中的进程数整除)。
- **group** (Group,可选) - 执行该操作的进程组实例(通过 ``new_group`` 创建)。默认为 None,即使用全局默认进程组。
- **sync_op** (bool,可选) - 该操作是否为同步操作。默认为 True,即同步操作。
- **use_calc_stream** (bool,可选) - 该操作是否在计算流上进行。默认为 False,即不在计算流上进行。该参数旨在提高同步操作的性能,请确保在充分了解其含义的情况下调整该参数的值。
Expand Down
16 changes: 8 additions & 8 deletions docs/api/paddle/incubate/nn/FusedDropoutAdd_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -9,26 +9,26 @@ FusedDropoutAdd

参数
:::::::::
- **p** (float|int, 可选): 将单位设置为零的概率。默认值: 0.5
- **mode** (str, 可选): ['upscale_in_train'(默认) | 'downscale_in_infer']
- **p** (float|int可选) - 将单位设置为零的概率。默认值0.5
- **mode** (str可选) - ['upscale_in_train'(默认)| 'downscale_in_infer']

1. upscale_in_train (默认), 在训练时放大输出
1. upscale_in_train(默认),在训练时放大输出

- train: :math:`out = x \times \frac{mask}{(1.0 - p)} + y`
- inference: :math:`out = x + y`

2. downscale_in_infer, 在推理时缩小输出
2. downscale_in_infer在推理时缩小输出

- train: :math:`out = x \times mask + y`
- inference: :math:`out = x \times (1.0 - p) + y`
- **name** (str, 可选): 操作的名称, 默认值为 None. 有关更多信息请参阅 :ref:`api_guide_Name`.
- **name** (str可选) - 操作的名称,默认值为 None有关更多信息请参阅 :ref:`api_guide_Name`


形状
:::::::::
- x: N-D 张量.
- y: N-D 张量.
- output: N-D 张量, 和 `x` 形状相同.
- x: N-D 张量
- y: N-D 张量
- output: N-D 张量和 `x` 形状相同


代码示例
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,23 +9,23 @@ fused_dropout_add

参数
:::::::::
- **x** (Tensor): 输入张量。数据类型为 bfloat16, float16, float32 或 float64.
- **y** (Tensor): 输入张量,数据类型为 bfloat16, float16, float32 或 float64.
- **p** (float|int, 可选): 将单位设置为零的概率。默认值: 0.5。
- **training** (bool, 可选): 标记是否为训练阶段。默认值: True。
- **mode** (str, 可选): ['upscale_in_train'(默认) | 'downscale_in_infer'].
- **x** (Tensor) - 输入张量。数据类型为 bfloat16float16float32 或 float64
- **y** (Tensor) - 输入张量,数据类型为 bfloat16float16float32 或 float64
- **p** (float|int可选) - 将单位设置为零的概率。默认值0.5。
- **training** (bool可选) - 标记是否为训练阶段。默认值True。
- **mode** (str可选) - ['upscale_in_train'(默认)| 'downscale_in_infer']

1. upscale_in_train (默认), 在训练时放大输出
1. upscale_in_train(默认),在训练时放大输出

- 训练: :math:`out = x \times \frac{mask}{(1.0 - dropout\_prob)} + y`
- 推理: :math:`out = x + y`

2. downscale_in_infer, 在推理时缩小输出
2. downscale_in_infer在推理时缩小输出

- 训练: :math:`out = input \times mask + y`
- 推理: :math:`out = input \times (1.0 - dropout\_prob) + y`

- **name** (str, 可选): 操作的名称, 默认值: None。有关更多信息, 请参阅 :ref:`api_guide_Name`.
- **name** (str可选) - 操作的名称,默认值None。有关更多信息请参阅 :ref:`api_guide_Name`


返回
Expand Down