cupy-numpy

y_cpu = cp.asnumpy(y_gpu)

y_gpu = cp.asarray(y_cpu)