Support GpuCumsum on 3D array.

Modified GpuCumsum op to support 3D array.
`a_shape` is only the initial shape, on the next line I change one of the axis size so it is not square any more. I put 3 instead of 5 to reduce unnecessary memory allocation.