mindspore.ops.cummin

mindspore.ops.cummin(x, axis)[源代码]

返回一个元组(最值、索引),其中最值是输入张量 x 沿维度 axis 的累积最小值,索引是每个最小值的索引位置。

\[\begin{split}\begin{array}{ll} \\ y{i} = min(x{1}, x{2}, ... , x{i}) \end{array}\end{split}\]

参数:

  • x (Tensor) - 输入Tensor,要求维度大于0。

  • axis (int) - 算子操作的维度,维度的大小范围是[-x.ndim, x.ndim - 1]。

返回:

一个包含两个Tensor的元组,分别表示累积最小值和对应索引。每个输出Tensor的形状和输入Tensor的形状相同。

异常:

  • TypeError - 如果 x 不是Tensor。

  • TypeError - 如果 axis 不是int。

  • ValueError - 如果 axis 不在范围[-x.ndim, x.ndim - 1]内。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor, ops
>>> import mindspore
>>> a = Tensor([-0.2284, -0.6628,  0.0975,  0.2680, -1.3298, -0.4220], mindspore.float32)
>>> output = ops.cummin(a, axis=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]