!28114 code_docs_update_cn_api_description

Merge pull request !28114 from zhuyuxiao/I4NR6X
This commit is contained in:
i-robot 2021-12-24 08:01:49 +00:00 committed by Gitee
commit b6d5648db7
1 changed files with 12 additions and 0 deletions

View File

@ -376,6 +376,18 @@
- **fusion_type** (int) Parameter的 `comm_fusion` 属性的设置值。
- **recurse** (bool) 是否递归地设置子Cell的可训练参数。默认值True。
.. py:method:: set_data_parallel()
递归设置该Cell中的所有算子的并行策略为数据并行。
.. note:: 仅在全自动并行(AUTO_PARALLEL)模式下生效。
**样例:**
>>> import mindspore.nn as nn
>>> net = nn.Dense(3, 4)
>>> net.set_data_parallel()
.. py:method:: set_grad(requires_grad=True)
Cell的梯度设置。在PyNative模式下该参数指定Cell是否需要梯度。如果为True则在执行正向网络时将生成需要计算梯度的反向网络。