tenferro-einsum 0.4.0

Subscripts, contraction planning, concrete/traced/eager einsum APIs, extension runtime, and AD rule for tenferro.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! Einsum and tensordot extension traits for concrete, eager, and traced values.

pub use crate::{
    ContractionTree, EinsumAxis, EinsumNotation, EinsumOptimize, EinsumSubscripts, TensorDotAxes,
    TensorEinsumExt, TensorEinsumIntoExt, TensorReadEinsumExt, TensorReadEinsumIntoExt,
    TensorTensordotExt, TraceContextEinsumExt, TracedTensorEinsumExt, TypedTensorEinsumExt,
    TypedTensorEinsumIntoExt, TypedTensorReadEinsumExt, TypedTensorReadEinsumIntoExt,
    TypedTensorTensordotExt,
};
pub use tenferro_runtime::{
    DType, Tensor, TensorBackend, TensorRead, TensorScalar, TensorView, TraceContext, TraceValue,
    TracedTensor, TypedTensor, TypedTensorView,
};
pub use tenferro_tensor::{DotGeneralAccumulation, TensorWrite, TypedTensorWrite};

#[cfg(feature = "autodiff")]
pub use crate::{EagerEinsumExt, EagerTensorEinsumExt};
#[cfg(feature = "autodiff")]
pub use tenferro_ad::{EagerRuntime, EagerTensor};