scikit-cuda icon indicating copy to clipboard operation
scikit-cuda copied to clipboard

skcuda.linalg.misc.multiply doesnt support 3D array broadcasting

Open itaysason opened this issue 6 years ago • 2 comments

`import pycuda.gpuarray as gpuarray import pycuda.autoinit import numpy as np import skcuda.linalg as linalg

linalg.init() a = np.arange(523).reshape(5,2,3).astype('float32') b = np.arange(2*3).reshape(2,3).astype('float32') a_gpu = gpuarray.to_gpu(a) b_gpu = gpuarray.to_gpu(b) a_gpu_times_b_gpu = linalg.misc.multiply(a_gpu, b_gpu)`

itaysason avatar Apr 23 '18 13:04 itaysason

Is there any function to multiply 3d matrices with broadcasting?

tugcetasci avatar Aug 16 '18 17:08 tugcetasci

At the present time, no.

lebedov avatar Aug 19 '18 21:08 lebedov