mindspore.ops.Cummin

查看源文件
class mindspore.ops.Cummin(axis)[源代码]

返回输入Tensor在指定轴上的累积最小值与其对应的索引。

警告

这是一个实验性API,后续可能修改或删除。

更多参考详见 mindspore.ops.cummin()

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

输入:
  • input (Tensor) - 输入Tensor。

输出:

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

支持平台:

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)
>>> axis = 0
>>> output = ops.Cummin(axis)(a)
>>> print(output[0])
[-0.2284 -0.6628 -0.6628 -0.6628 -1.3298 -1.3298]
>>> print(output[1])
[0 1 1 1 4 4]