take_along_axis_kernel.cu 2.2 KB