mindspore.ops.Median

class mindspore.ops.Median(global_median=False, axis=0, keep_dims=False)[源代码]

输出Tensor指定维度 axis 上的中值与其对应的索引。如果 global_median 为True，则计算Tensor中所有元素的中值。

警告

如果 global_median 为True, 第二个输出 indices 无意义。

参数：

global_median (bool, 可选) - 是否计算Tensor中所有元素的中值。默认值：False。
axis (int, 可选) - 进行中值计算的轴。默认值：0。
keep_dims (bool, 可选) - 是否保留 axis 指定的维度。默认值：False。

输入：

x (Tensor) - 要计算中值的Tensor。支持的数据类型：int16、int32、int64、float32或float64。

输出：

y (Tensor) - 中值，数据类型与 x 相同。
- 如果 global_median 为True， y 只有一个元素。
- 如果 keep_dims 为True, y 的shape除了在 axis 维度上为1外与 x 一致。
- 其他情况下， y 比 x 缺少 axis 指定的维度。
indices (Tensor) - 中值的索引。shape与 y 一致，数据类型为int64。

异常：

TypeError - x 的数据类型不是: int16、int32、int64、float32、float64。
TypeError - x 不是Tensor。
TypeError - global_median 或 keep_dims 被指定了一个非bool值。
TypeError - axis 不是int。
ValueError - axis 不在 [-x.dim, x.dim-1] 范围内。

支持平台：

GPU CPU

样例：

>>> # case 1 : common median compute
>>> from mindspore import Tensor, ops
>>> import numpy as np
>>> x = Tensor(np.array([[5, 1, 2],[3, 5, 7], [1, 6, 4]]).astype(np.int64))
>>> median = ops.Median(global_median=False, axis=0, keep_dims=False)
>>> y = median(x)
>>> print(y)
(Tensor(shape=[3], dtype=Int64, value= [3, 5, 4]), Tensor(shape=[3], dtype=Int64, value= [1, 1, 2]))
>>> # case 2 : global median compute
>>> from mindspore import Tensor, ops
>>> import numpy as np
>>> x = Tensor(np.array([[1, 7, 6],[5, 1, 3],[9, 17, 1]]).astype(np.int32))
>>> median = ops.Median(global_median=True)
>>> y = median(x)
>>> print(y)
(Tensor(shape=[], dtype=Int32, value= 5), Tensor(shape=[], dtype=Int64, value= 0))