Is this different than `np.apply_along_axis`?