Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add prod in reduce note

This commit is contained in:
wangshengnan123 2023-02-02 09:31:51 +08:00
parent c342c68673
commit 1d5f0f52b9
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/api/api_python/ops/mindspore.ops.AllReduce.rst
View File

@ -9,7 +9,7 @@
集合中的所有进程的Tensor必须具有相同的shape和格式。
参数：
- **op** (str) - 规约的具体操作，如"sum"、"max"、和"min"。默认值：ReduceOp.SUM。
- **op** (str) - 规约的具体操作，如"sum"、"prod"、"max"、和"min"。默认值：ReduceOp.SUM。
- **group** (str) - 工作的通信组。默认值："GlobalComm.WORLD_COMM_GROUP"（即Ascend平台为"hccl_world_group"，GPU平台为"nccl_world_group" ）。
输入：

2
mindspore/python/mindspore/ops/operations/comm_ops.py
View File

@ -117,7 +117,7 @@ class AllReduce(Primitive):
The tensors must have the same shape and format in all processes of the collection.
Args:
op (str): Specifies an operation used for element-wise reductions, like sum, max, and min.
op (str): Specifies an operation used for element-wise reductions, like sum, prod, max, and min.
On the CPU, only 'sum' is supported. Default: ReduceOp.SUM.
group (str): The communication group to work on. Default: "GlobalComm.WORLD_COMM_GROUP".