modify apiformat

This commit is contained in:
xumengjuan1 2022-03-07 15:19:02 +08:00
parent 8767a804f2
commit 1ebb69de3f
4 changed files with 4 additions and 1 deletions

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

@ -28,5 +28,6 @@ mindspore.ops.NotEqual
Tensor输出shape与输入相同数据类型为bool。
**异常:**
- **TypeError** - `x``y` 不是以下之一Tensor、Number、bool。
- **TypeError** - `x``y` 都不是Tensor。

View File

@ -83,7 +83,7 @@ mindspore.ops.Primitive
- **stage** (int) - 当前stage的ID。
.. py:method:: shard(in_strategy, out_strategy)
.. py:method:: shard(in_strategy=None, out_strategy=None)
将切分策略添加到Primitive属性中。

View File

@ -168,6 +168,7 @@ class Primitive(Primitive_):
in_strategy (tuple): Describe the split strategy of operator input. Default: None.
out_strategy (tuple): Describe the split strategy of operator output, it is only for certain operators,
such as MatMul. Default: None.
Examples:
>>> from mindspore import ops
>>> add = ops.Add()