Skip to content

Commit

Permalink
update docs for broadcast,broadcast_tensors,multiplex,unbind (#6331)
Browse files Browse the repository at this point in the history
  • Loading branch information
zyt1024 authored Nov 29, 2023
1 parent d45d563 commit 8b0d91f
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion docs/api/paddle/broadcast_tensors_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ broadcast_tensors

参数
:::::::::
- inputs (list(Tensor)|tuple(Tensor)) - 一组输入 Tensor,数据类型为:bool、float32、float64、int32 或 int64。所有的输入 Tensor 均需要满足 rank <= 5。
- inputs (list(Tensor)|tuple(Tensor)) - 一组输入 Tensor,数据类型为:bool、float32、float64、complex64、complex128、int32 或 int64。所有的输入 Tensor 均需要满足 rank <= 5。
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。

返回
Expand Down
2 changes: 1 addition & 1 deletion docs/api/paddle/distributed/broadcast_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ broadcast

参数
:::::::::
- **tensor** (Tensor) - 在目标进程上为待广播的 tensor,在其他进程上为用于接收广播结果的 tensor。支持的数据类型包括:float16、float32、float64、int32、int64、int8、uint8、bool、bfloat16。
- **tensor** (Tensor) - 在目标进程上为待广播的 tensor,在其他进程上为用于接收广播结果的 tensor。支持的数据类型包括:float16、float32、float64、int32、int64、int8、uint8、bool、bfloat16、complex64、complex128
- **src** (int) - 目标进程的 rank,该进程传入的 tensor 将被发送到其他进程上。
- **group** (Group,可选) - 执行该操作的进程组实例(通过 ``new_group`` 创建)。默认为 None,即使用全局默认进程组。
- **sync_op** (bool,可选) - 该操作是否为同步操作。默认为 True,即同步操作。
Expand Down
2 changes: 1 addition & 1 deletion docs/api/paddle/multiplex_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ multiplex
参数
::::::::::::

- **inputs** (list) - 为输入 Tensor 列表,列表元素为数据类型为 float32、float64、int32、int64 的多维 Tensor。所有输入 Tensor 的 shape 应相同,秩必须至少为 2。
- **inputs** (list) - 为输入 Tensor 列表,列表元素为数据类型为 float32、float64、int32、int64、complex64、complex128 的多维 Tensor。所有输入 Tensor 的 shape 应相同,秩必须至少为 2。
- **index** (Tensor)- 用来选择输入 Tensor 中的某些行构建输出 Tensor 的索引,为数据类型为 int32 或 int64、shape 为[M, 1]的 2-D Tensor,其中 M 为输入 Tensor 个数。
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。

Expand Down
2 changes: 1 addition & 1 deletion docs/api/paddle/unbind_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ unbind

参数
:::::::::
- **input** (Tensor) - 输入变量,数据类型为 float16、loat32、float64、int32、int64 的多维 Tensor。
- **input** (Tensor) - 输入变量,数据类型为 float16、loat32、float64、int32、int64、complex64、complex128 的多维 Tensor。
- **axis** (int32|int64,可选) - 数据类型为 int32 或 int64,表示需要分割的维度。如果 axis < 0,则划分的维度为 rank(input) + axis。默认值为 0。

返回
Expand Down

0 comments on commit 8b0d91f

Please sign in to comment.