cupy.median#

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

计算沿指定轴的中位数。

返回数组元素的中位数。

参数:

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

  • axis (int, sequence of int or None) – 计算中位数所沿的轴。默认使用展平的数组。

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

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

  • keepdims (bool) – 如果 True，则轴将保持为大小为1的轴。

返回:

a 的中位数，如果指定了轴。

返回类型:

cupy.ndarray

参见

numpy.median()