take_along_axis_grad_kernel.cu 2.7 KB