Tag: apply-along-axis
Handling dtype conflicts when using np.apply_along_axis with custom functions