[Numpy-discussion] tensor contractions

Gideon Simpson simpson at math.toronto.edu
Fri Jan 16 22:14:16 EST 2009


Suppose I have a 3d array, A, with dimensions 2 x 2 x N, and a 2d  2 x  
N array, u.  I interpret A as N 2x2 matrices and u as N 2d vectors.   
Suppose I want to apply the mth matrix to the mth vector, i.e.

A[, , m] u[, m] = v[, m]

Aside from doing
A[0,0,:] u[0,:]  + A[0,1,:] u[1,:] = v[0,:]
and
A[1,0,:] u[0,:]  + A[1,1,:] u[1,:] = v[1,:]
is there a smart way to perform this computation?

-gideon




More information about the NumPy-Discussion mailing list