scikit-cuda
scikit-cuda copied to clipboard
skcuda.linalg.misc.multiply doesnt support 3D array broadcasting
`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)`
Is there any function to multiply 3d matrices with broadcasting?
At the present time, no.