[Numpy-discussion] Matrix dot product over an axis(for a 3d array/list of matrices)

David Warde-Farley dwf at cs.toronto.edu
Thu Jul 15 18:28:39 EDT 2010


On 2010-07-15, at 4:31 PM, David Warde-Farley wrote:

> If you need/want more speed than the solution Chuck proposed, you should check out Cython and Tokyo. Cython lets you write loops that execute at C speed, whereas Tokyo provides a Cython level wrapper for BLAS (no need to go through Python code to call NumPy). Tokyo was designed for exactly your use case: lots of matrix multiplies with relatively small matrices, where you start noticing the Python overhead.

It occurred to me I neglected to provide a link (cursed iPhone):

http://www.vetta.org/2009/09/tokyo-a-cython-blas-wrapper-for-fast-matrix-math/

David


More information about the NumPy-Discussion mailing list