!30838 modify apiformat

Merge pull request !30838 from xumengjuan1/code_docs_xx8
This commit is contained in:
i-robot 2022-03-04 07:52:03 +00:00 committed by Gitee
commit 415d46d2db
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
13 changed files with 23 additions and 14 deletions

View File

@ -24,7 +24,7 @@ Boost能够自动加速网络如减少BN/梯度冻结/累积梯度等。
**输出:**
Tuple(Tensor), adasum处理后更新的权重。
- **Tuple** (Tensor) - adasum处理后更新的权重。
.. py:class:: mindspore.boost.AutoBoost(level="O0", boost_config_dict="")
@ -169,8 +169,8 @@ Boost能够自动加速网络如减少BN/梯度冻结/累积梯度等。
**参数:**
- network (Cell)训练网络。
- optimizer (Union[Cell])用于更新权重的优化器。
- **network** (Cell) - 训练网络。
- **optimizer** (Union[Cell]) - 用于更新权重的优化器。
.. py:method:: network_auto_process_eval(network)
@ -178,7 +178,7 @@ Boost能够自动加速网络如减少BN/梯度冻结/累积梯度等。
**参数:**
network(Cell)推理网络。
**network** (Cell) - 推理网络。
.. py:class:: mindspore.boost.BoostTrainOneStepCell(network, optimizer, sens=1.0)

View File

@ -314,9 +314,11 @@
- **hook_fn** (function) 捕获Cell对象信息和正向输入数据的hook_fn函数。
**返回:**
- **handle** 与hook_fn函数对应的handle对象。
**异常:**
- **TypeError** 如果 `hook_fn` 不是Python函数。
.. py:method:: register_forward_hook(hook_fn)
@ -333,9 +335,11 @@
- **hook_fn** (function) 捕获Cell对象信息和正向输入输出数据的hook_fn函数。
**返回:**
- **handle** 与hook_fn函数对应的handle对象。
**异常:**
- **TypeError** 如果 `hook_fn` 不是Python函数。
.. py:method:: register_backward_hook(hook_fn)
@ -352,9 +356,11 @@
- **hook_fn** (function) 捕获Cell对象信息和反向输入输出梯度的hook_fn函数。
**返回:**
- **handle** 与hook_fn函数对应的handle对象。
**异常:**
- **TypeError** 如果 `hook_fn` 不是Python函数。
.. py:method:: remove_redundant_parameters()

View File

@ -1,7 +1,7 @@
mindspore.nn.ConfusionMatrix
============================
.. py:class:: mindspore.nn.ConfusionMatrix(num_classes, normalize='NO_NORM', threshold=0.5)
.. py:class:: mindspore.nn.ConfusionMatrix(num_classes, normalize='no_norm', threshold=0.5)
计算混淆矩阵(confusion matrix),通常用于评估分类模型的性能,包括二分类和多分类场景。

View File

@ -36,6 +36,7 @@ mindspore.nn.GRU
**输出:**
Tuple包含(`output`, `h_n`)的tuple。
- **output** (Tensor) - shape为(seq_len, batch_size, num_directions * `hidden_size`)或(batch_size, seq_len, num_directions * `hidden_size`)的Tensor。
- **hx_n** (Tensor) - shape为(num_directions * `num_layers`, batch_size, `hidden_size`)的Tensor。

View File

@ -29,7 +29,7 @@ mindspore.nn.GRUCell
**输出:**
- **h'** (Tensor) - shape为(batch_size, `hidden_size`)的Tensor。
- **hx** (Tensor) - shape为(batch_size, `hidden_size`)的Tensor。
**异常:**

View File

@ -44,7 +44,7 @@ mindspore.nn.HausdorffDistance
**参数:**
- inputs`y_pred``y``label_idx``y_pred``y` 为Tensor list或numpy.ndarray`y_pred` 是预测的二值图像,`y` 是实际的二值图像。`label_idx` 的数据类型为int或float表示像素点的类别值。
- **inputs** - `y_pred``y``label_idx``y_pred``y` 为Tensor list或numpy.ndarray`y_pred` 是预测的二值图像,`y` 是实际的二值图像。`label_idx` 的数据类型为int或float表示像素点的类别值。
**异常:**

View File

@ -13,11 +13,11 @@ mindspore.nn.L1Regularizer
**参数:**
- **scale (int, float)** - L1正则化因子其值大于0。
- **scale** (int, float) - L1正则化因子其值大于0。
**输入:**
- **weights** (Tensor)** - L1Regularizer的输入任意维度的Tensor数据类型为float16或float32。
- **weights** (Tensor) - L1Regularizer的输入任意维度的Tensor数据类型为float16或float32。
**输出:**

View File

@ -20,6 +20,7 @@ mindspore.nn.LSTMCell
其中 :math:`\sigma` 是sigmoid函数 :math:`*` 是乘积。 :math:`W,b` 是公式中输出和输入之间的可学习权重。例如,:math:`W_{ix}, b_{ix}` 是用于从输入 :math:`x` 转换为 :math:`i` 的权重和偏置。详见论文 `LONG SHORT-TERM MEMORY <https://www.bioinf.jku.at/publications/older/2604.pdf>`_`Long Short-Term Memory Recurrent Neural Network Architectures for Large Scale Acoustic Modeling <https://static.googleusercontent.com/media/research.google.com/zh-CN//pubs/archive/43905.pdf>`_
**参数:**
- **input_size** (int) - 输入的大小。
- **hidden_size** (int)- 隐藏状态大小。
- **has_bias** (bool) - cell是否有偏置 `b_ih``b_hh` 。默认值True。

View File

@ -26,6 +26,7 @@
**输出:**
tuple (Tensor)表示2个Tensor组成的tuple包含对应的索引和输入Tensor的最大值。
- **index** (Tensor) - 输入Tensor最大值的索引。如果 `keep_dims` 为true则输出Tensor的shape为 :math:`(x_1, x_2, ..., x_{axis-1}, 1, x_{axis+1}, ..., x_N)` 。否则shape为 :math:`(x_1, x_2, ..., x_{axis-1}, x_{axis+1}, ..., x_N)`
- **output_x** (Tensor) - 输入Tensor的最大值其shape与索引相同。

View File

@ -9,7 +9,7 @@
**参数:**
num_true (int)每个训练样本的目标类数。默认值1。
- **num_true** (int)每个训练样本的目标类数。默认值1。
**输入:**

View File

@ -13,7 +13,7 @@
**参数:**
keep_prob (float)输入通道保留率数值范围在0到1之间例如 `keep_prob` = 0.8意味着过滤20%的通道。默认值0.5。
- **keep_prob** (float)输入通道保留率数值范围在0到1之间例如 `keep_prob` = 0.8意味着过滤20%的通道。默认值0.5。
**输入:**

View File

@ -41,5 +41,5 @@ mindspore.ops.NLLLoss
**异常:**
- ** TypeError** - `logits``weight` 的数据类型既不是float16也不是float32 `labels` 不是int32。
- ** ValueError** - `logits` 不是一维或二维Tensor `labels``weight` 不是一维Tensor。 `logits` 是二维Tensor时 `logits` 的第一个维度不等于 `labels` `logits` 的第二个维度不等于 `weight``logits` 是一维Tensor时 `logits``labels``weight` 的维度应该相同。
- **TypeError** - `logits``weight` 的数据类型既不是float16也不是float32 `labels` 不是int32。
- **ValueError** - `logits` 不是一维或二维Tensor `labels``weight` 不是一维Tensor。 `logits` 是二维Tensor时 `logits` 的第一个维度不等于 `labels` `logits` 的第二个维度不等于 `weight``logits` 是一维Tensor时 `logits``labels``weight` 的维度应该相同。

View File

@ -14,7 +14,7 @@
**输入:**
- **input_ids** (Tensor) - shape为(batch_size, seq_length)的输入其数据类型为int32。
**input_ids** (Tensor) - shape为(batch_size, seq_length)的输入其数据类型为int32。
**输出:**