如何在pycuda / scikits.cuda中进行元素级分配？
如何在pycuda / scikits.cuda中进行元素级分配？
提问于 2014-11-15 01:07:54
下面是代码：
import pycuda.autoinit
import pycuda.gpuarray as gpuarray
import pycuda.driver as drv
import numpy as np
import scikits.cuda.linalg as culinalg
import scikits.cuda.misc as cumisc
culinalg.init()
ag = gpuarray.to_gpu(np.random.rand(1000,1000))
bg = gpuarray.to_gpu(np.zeros((1000,1000))
bg[:,:] = ag我得到了以下错误：
TypeError: 'GPUArray' object does not support item assignment那么，在pycuda/ scikits.cuda中将一个矩阵分配给另一个现有矩阵的方法有哪些？
回答 1
Stack Overflow用户
回答已采纳
发布于 2014-11-15 08:38:03
好的，这不是一种优雅的方法，而是一种解决方案:使用pycuda的ElementwiseKernel：
import from pycuda.elementwise import ElementwiseKernel
ele_assign = ElementwiseKernel("double *a,double *b","a[i] = b[i]","ele_assign")
ele_assign(bg,ag)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/26941371
复制相关文章
相似问题
腾讯云开发者
