[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