forked from mindspore-Ecosystem/mindspore
25 lines
1.2 KiB
ReStructuredText
25 lines
1.2 KiB
ReStructuredText
|
mindspore.ops.aminmax
|
|||
|
|
======================
|
|||
|
|
|
|||
|
|
.. py:function:: mindspore.ops.aminmax(x, *, axis=0, keep_dims=False)
|
|||
|
|
|
|||
|
|
返回输入Tensor在指定轴上的最小值和最大值。
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
- **x** (Tensor) - 输入Tensor，可以是任意维度。设输入Tensor的shape为 :math:`(x_1, x_2, ..., x_N)` 。
|
|||
|
|
|
|||
|
|
关键字参数：
|
|||
|
|
- **axis** (int，可选) - 要进行规约计算的维度。 `axis` 必须在[-rank, rank)范围内，其中 “rank” 是 `x` 的维度。默认值：0。
|
|||
|
|
- **keep_dims** (bool，可选) - 是否保留维度。如果为True，则输出shape与输入shape一致，否则移除规约计算的维度 `axis` 。默认值：False。
|
|||
|
|
|
|||
|
|
返回：
|
|||
|
|
tuple (Tensor)，包含输入Tensor在指定轴上的最小值和最大值。
|
|||
|
|
|
|||
|
|
- `keepdims` 为True，输出shape为： :math:`(x_1, x_2, ..., x_{axis-1}, 1, x_{axis+1}, ..., x_N)` 。
|
|||
|
|
- `keepdims` 为False，输出shape为： :math:`(x_1, x_2, ..., x_{axis-1}, x_{axis+1}, ..., x_N)` 。
|
|||
|
|
|
|||
|
|
异常：
|
|||
|
|
- **TypeError** - `keepdims` 不是bool类型。
|
|||
|
|
- **TypeError** - `axis` 不是int类型。
|
|||
|
|
- **ValueError** - `axis` 不在[-rank, rank)范围内。
|