[Numpy-discussion] Matrix products

konrad.hinsen at laposte.net konrad.hinsen at laposte.net
Tue Jan 25 05:33:01 EST 2005


On Jan 25, 2005, at 14:10, Curzio Basso wrote:

> assume that I have a matrix A with shape = (m,n), what I would like to 
> compute is a matrix B with shape = (m, n, n) such as
> B[i] = NA.matrixmultiply(A[i, :, NA.NewAxis], A[i, NA.NewAxis])
...
> Does anyone know how to do this without using loops?

How about

	A[:, :, NewAxis]*A[:, NewAxis, :]

That works for your example at least. I am not quite sure why you use 
matrixmultiply in your definition as there doesn't seem to be any 
summation.

Konrad.
--
---------------------------------------------------------------------
Konrad Hinsen
Laboratoire Léon Brillouin, CEA Saclay,
91191 Gif-sur-Yvette Cedex, France
Tel.: +33-1 69 08 79 25
Fax: +33-1 69 08 82 61
E-Mail: hinsen at llb.saclay.cea.fr
---------------------------------------------------------------------





More information about the NumPy-Discussion mailing list