ENH: support ufunc keepdims argument in np.median