papyri
{ } Raw JSON

bundles / scipy latest / scipy / integrate / _bvp / stacked_matmul

function

scipy.integrate._bvp:stacked_matmul

source: /scipy/integrate/_bvp.py :144

Signature

def   stacked_matmul ( a ,  b )

Summary

Stacked matrix multiply: out[i,:,:] = np.dot(a[i,:,:], b[i,:,:]).

Extended Summary

Empirical optimization. Use outer Python loop and BLAS for large matrices, otherwise use a single einsum call.

Aliases

  • scipy.integrate._bvp.stacked_matmul