You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
556 B
28 lines
556 B
13 years ago
|
# $Id: __init__.py,v 1.1 2011-07-14 19:00:59 wirawan Exp $
|
||
|
|
#
|
||
|
|
# wpylib.math.linalg main module
|
||
|
|
# Created: 20110714
|
||
|
|
# Wirawan Purwanto
|
||
|
|
#
|
||
|
|
"""
|
||
|
|
wpylib.math.linalg
|
||
|
|
|
||
|
|
Provides convenience functions for linear algebra things beyond what's
|
||
|
|
already provided by numpy.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
import numpy
|
||
|
|
|
||
|
|
def matmul(*Mats):
|
||
|
|
"""Do successive matrix product. For example,
|
||
|
|
matmul(A,B,C,D)
|
||
|
|
will evaluate a matrix multiplication ((A*B)*C)*D .
|
||
|
|
The matrices must be of matching sizes."""
|
||
|
|
p = numpy.dot(Mats[0], Mats[1])
|
||
|
|
for M in Mats[2:]:
|
||
|
|
p = numpy.dot(p, M)
|
||
|
|
return p
|
||
|
|
|
||
|
|
|