Module einsum2 

Binary einsum contractions on strided views.

Modules

backend

Backend abstraction for batched GEMM dispatch. Backend abstraction for batched GEMM dispatch.

bgemm_faer

Batched GEMM backend using the [faer] library. faer-backed batched GEMM kernel on strided views.

bgemm_naive

Batched GEMM fallback using explicit loops. Naive batched GEMM kernel on strided views.

contiguous

GEMM-ready operand types and preparation functions for contiguous data. GEMM-ready operand types and preparation functions for contiguous data.

plan

Contraction planning: axis classification and permutation computation. Einsum2 plan: axis classification and permutation computation.

trace

Trace-axis reduction (summing axes that appear only in one operand). Trace-axis reduction for einsum operands.

util

Shared helpers (permutation inversion, multi-index iteration, dimension fusion). Shared helpers for strided-einsum2.

Structs

Einsum2Plan

Pre-computed execution plan for a binary einsum contraction.

Enums

EinsumError

Errors specific to einsum operations.

Traits

AxisId

Trait alias for axis label types.

Backend

Trait for backends that can execute batched GEMM on contiguous operands.

Scalar

Trait alias for element types supported by einsum operations.

ScalarBase

Shared trait bounds for all element types usable with einsum, independent of GEMM backend.

Functions

einsum2_into

Binary einsum contraction: C = alpha * contract(A, B) + beta * C.

einsum2_into_owned

Binary einsum accepting owned inputs for zero-copy optimization.

einsum2_naive_into

Binary einsum for custom scalar types using naive GEMM.

einsum2_with_backend_into

Binary einsum with a pluggable GEMM backend.

Type Aliases

Result

Convenience alias for Result<T, EinsumError>.