On Thu, Jul 15, 2010 at 12:00 PM, Emmanuel Bengio <beng...@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
_______________________________________________
NumPy-Discussion mailing list
NumPy-Discussion@scipy.org
http://mail.scipy.org/mailman/listinfo/numpy-discussion

Reply via email to