来看几个例子:
import numpy as np
import cupy as cp
import time
x=np.ones((1024,512,4,4))*1024.
y=np.ones((...后面的计算速度才是稳定的,cpu和gpu都有一定这个特性,这个原因cpu和gpu是不同!和“操作系统的本身算法、GPU工作方式”等有关系吧?...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’就好了!