Function apply_matrix

dedalus/tools/array.py:77–82

Apply matrix along any axis of an array.

(matrix, array, axis, **kw)

def apply_matrix(matrix, array, axis, **kw):
    """Apply matrix along any axis of an array."""
    if sparse.isspmatrix(matrix):
        return apply_sparse(matrix, array, axis, **kw)
    else:
        return apply_dense(matrix, array, axis, **kw)


def apply_dense_einsum(matrix, array, axis, optimize=True, **kw):