fix: make sure 'at' is a cupy zero dim array

fix: make sure 'at' is a cupy zero dim array #6477