cupy.median#

cupy.median(a, axis=None, out=None, overwrite_input=False, keepdims=False)[源代码]#

计算指定轴上的中位数。

返回数组元素的中位数。

参数:

  • a (cupy.ndarray) – 要计算中位数的数组。

  • axis (int, ints 序列 或 None) – 计算中位数的轴。默认使用扁平化的数组。

  • out (cupy.ndarray) – 输出数组。

  • overwrite_input (bool) – 如果为 True，则允许使用输入数组 a 的内存进行计算。输入数组将被 median 调用修改。当您不需要保留输入数组的内容时，这将节省内存。将输入视为未定义，但它可能会被完全或部分排序。默认为 False。如果 overwrite_input 为 True 且 a 尚未是 ndarray，则会引发错误。

  • keepdims (bool) – 如果为 True，则该轴将保留为大小为一的轴。

返回:

a 的中位数，如果指定了轴，则返回沿轴方向的中位数。

返回类型:

cupy.ndarray

另请参阅

numpy.median()