用pycuda实现numpy.argwhere函数处理三维数组