From 51dc15b4b5df7e544c7f196796e716aaedb6cadd Mon Sep 17 00:00:00 2001 From: zhuyuxiao Date: Fri, 24 Dec 2021 09:38:31 +0800 Subject: [PATCH] update cn api description --- docs/api/api_python/nn/mindspore.nn.Cell.rst | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/docs/api/api_python/nn/mindspore.nn.Cell.rst b/docs/api/api_python/nn/mindspore.nn.Cell.rst index 3384894e265..58a491a8361 100644 --- a/docs/api/api_python/nn/mindspore.nn.Cell.rst +++ b/docs/api/api_python/nn/mindspore.nn.Cell.rst @@ -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,则在执行正向网络时,将生成需要计算梯度的反向网络。