tenferro-einsum 0.3.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

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

pub use crate::{
    ContractionTree, 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};