Adds a vector to the diagonal of an array.
This is the same as A + diag(D), but faster.
Parameters: | d_gpu : pycuda.gpuarray.GPUArray
a_gpu : pycuda.gpuarray.GPUArray
overwrite : bool
handle : int
|
---|---|
Returns: | r_gpu : pycuda.gpuarray.GPUArray
|
Notes
d_gpu and a_gpu must have the same precision data type.