take_along_axis_kernel.cc 2.0 KB