take_along_axis_op.cc 5.9 KB