来看几个例子:
import numpy as np
import cupy as cp
import time
x=np.ones((1024,512,4,4))*1024.
y=np.ones((...4,4,4,1))*512.3254
x=cp.ones((4,4,4,4))*1024.
y=cp.ones((4,4,4,1))*512.3254
GPU失去了优势,所以也不是所有计算都需要放到gpu上来加速的...cupy种几乎包含了numpy种通常有的很多function了!所以基本上再用的时候只要把‘np’ 换成‘cp’就好了!