線性代數與 np.linalg

從版本 1.8 開始，np.linalg 中的幾個例程可以在矩陣的堆疊上執行。也就是說，例程可以計算多個矩陣的結果，如果它們堆疊在一起的話。例如，A 在這裡被解釋為兩個堆疊的 3×3 矩陣：

np.random.seed(123)
A = np.random.rand(2,3,3)
b = np.random.rand(2,3)
x = np.linalg.solve(A, b)

print np.dot(A[0,:,:], x[0,:])
# array([ 0.53155137,  0.53182759,  0.63440096])

print b[0,:]
# array([ 0.53155137,  0.53182759,  0.63440096])

官方 np 文件通過像 a : (..., M, M) array_like 這樣的引數規範來指定。