mindspore.mint.cummin
- mindspore.mint.cummin(input, dim)[源代码]
返回一个元组(最值、索引),其中最值是输入Tensor input 沿维度 dim 的累积最小值,索引是每个最小值的索引位置。
\[\begin{split}\begin{array}{ll} \\ y_{i} = \min(x_{1}, x_{2}, ... , x_{i}) \end{array}\end{split}\]- 参数:
input (Tensor) - 输入Tensor,要求维度大于0。
dim (int) - 算子操作的维度,维度的大小范围是[-input.ndim, input.ndim - 1]。
- 返回:
一个包含两个Tensor的元组,分别表示累积最小值和对应索引。每个输出Tensor的形状和输入Tensor的形状相同。
- 异常:
TypeError - 如果 input 不是Tensor。
TypeError - 如果 input 的Tensor类型是复数或bool。
TypeError - 如果 dim 不是int。
ValueError - 如果 dim 不在范围[-input.ndim, input.ndim - 1]内。
说明
Ascend不支持O2模式。
- 支持平台:
Ascend
样例:
>>> from mindspore import Tensor, mint >>> import mindspore >>> a = Tensor([-0.2284, -0.6628, 0.0975, 0.2680, -1.3298, -0.4220], mindspore.float32) >>> output = mint.cummin(a, dim=0) >>> print(output[0]) [-0.2284 -0.6628 -0.6628 -0.6628 -1.3298 -1.3298] >>> print(output[1]) [0 1 1 1 4 4]