op_take_along_axis function

Select values from x at the 1-D indices along the given axis.