mindspore/docs/api/api_python/mint/mindspore.mint.func_cummax.rst

23 lines
517 B
ReStructuredText

mindspore.mint.cummax
======================
.. py:function:: mindspore.mint.cummax(input, dim)
返回tensor在指定维度上的累积最大值及其索引。
.. math::
\begin{array}{ll} \\
y_{i} = \max(x_{1}, x_{2}, ... , x_{i})
\end{array}
.. note::
Ascend不支持GE后端。
参数:
- **input** (Tensor) - 输入tensor。
- **dim** (int) - 指定计算的维度。
返回:
两个tensor组成的tuple(max, max_indices)。