mindspore.mint.median

mindspore.mint.median(input, dim=None, keepdim=False)

输出指定维度上的中值与其对应的索引。

参数：

• input (Tensor) - 输入tensor。

• dim (int, 可选) - 指定计算维度。默认 None 。

• keepdim (bool, 可选) - 输出tensor是否保留维度。默认 False 。

返回：

两个tensor组成的tuple(median, median_indices)

支持平台：

Ascend

样例：

>>> import numpy as np
>>> import mindspore
>>> x = mindspore.tensor(np.array([[0.57, 0.11, 0.21],[0.38, 0.50, 0.57], 
...                                [0.36, 0.16, 0.44]]).astype(np.float32))
>>> y = mindspore.mint.median(x, dim=0, keepdim=False)
>>> print(y)
(Tensor(shape=[3], dtype=Float32, value= [ 3.79999995e-01,  1.59999996e-01,  4.39999998e-01]),
Tensor(shape=[3], dtype=Int64, value= [1, 2, 2]))