forked from huawei/mindspore2022
modify apiformat
This commit is contained in:
parent
8767a804f2
commit
1ebb69de3f
|
|
@ -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与索引相同。
|
||||
|
||||
|
|
|
|||
|
|
@ -28,5 +28,6 @@ mindspore.ops.NotEqual
|
|||
Tensor,输出shape与输入相同,数据类型为bool。
|
||||
|
||||
**异常:**
|
||||
|
||||
- **TypeError** - `x` 和 `y` 不是以下之一:Tensor、Number、bool。
|
||||
- **TypeError** - `x` 和 `y` 都不是Tensor。
|
||||
|
|
@ -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属性中。
|
||||
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Reference in New Issue