mindspore.mint.min
- mindspore.mint.min(input, dim=None, keepdim=False)[源代码]
在给定轴上计算输入Tensor的最小值。
- 参数:
input (Tensor) - 输入任意维度的Tensor。不支持complex类型。
dim (int, 可选) - 指定计算维度。默认值:
None
。keepdim (bool, 可选) - 表示是否减少维度,如果为
True
,输出将与输入保持相同的维度;如果为False
,输出将减少维度。默认值:False
。
- 返回:
Tensor,如果 dim 为默认值
None
,返回输入Tensor中所有元素的最小值,输出的shape为 \(()\) ,数据类型与 input 相同。tuple (Tensor),如果 dim 不为默认值
None
,表示2个Tensor组成的tuple,包含输入Tensor沿给定维度的最小值和对应的索引。values (Tensor) - 输入Tensor沿给定维度的最小值,数据类型和 input 相同。如果 keepdim 为
True
,输出Tensor的维度是 \((input_1, input_2, ...,input_{axis-1}, 1, input_{axis+1}, ..., input_N)\) 。否则输出维度为 \((input_1, input_2, ...,input_{axis-1}, input_{axis+1}, ..., input_N)\) 。index (Tensor) - 输入Tensor沿给定维度最小值的索引,shape和 values 相同。
- 异常:
ValueError - 如果 dim 为默认值
None
且 keepdim 不是False
。
- 支持平台:
Ascend
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, mint >>> x = Tensor(np.array([0.0, 0.4, 0.6, 0.7, 0.1]), mindspore.float32) >>> output, index = mint.min(x, 0, keepdim=True) >>> print(output, index) [0.0] [0]