papyri
{ } Raw JSON

bundles / scipy 1.18.0 / 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