pycuda 的 gpuarray.dot 计算异常问题~~~ 出来的是一个值
问题:
a = gpuarray.to_gpu(np.array(features1)) # [5,2]
b = gpuarray.to_gpu(np.array([features2]).T) #[2,1]
gpuarray.dot(a,b)
<<< 0.46748456456
<<< 实际上 应该是个 [5,1]维度的数组
解决方法:
方法1
for features in features1:
gpuarray.dot(features,b)
方法
原创
2022-03-10 19:14:01 ·
152 阅读 ·
0 评论