On Thu, Jul 15, 2010 at 12:00 PM, Emmanuel Bengio
<bengioe@gmail.com> wrote:
Ok I get it. Thanks!
Numpy syntax that works for me:
numpy.sum(a[:,:,:,numpy.newaxis]*b[:,numpy.newaxis,:,:],axis=-2)
The leading "..." gives the same thing, but iterates over all the leading indicies in case you want multidimensional arrays of matrices ;) You can also use the sum method which might be a bit more economical:
(a[:,:,:,numpy.newaxis]*b[:,numpy.newaxis,:,:]).sum(axis=-2)
How do the execution times compare?
<snip>
Chuck