Skip to main content

Crate trueno_tensor

Crate trueno_tensor 

Source
Expand description

N-dimensional tensor contractions — Einstein summation via TTGT.

This crate provides a dense tensor type and Einstein summation (einsum) that reduces tensor contractions to explicit loops, following the TTGT (Transpose-Transpose-GEMM-Transpose) strategy for cuTENSOR parity.

Re-exports§

pub use einsum::batch_matmul;
pub use einsum::einsum;
pub use einsum::einsum_nary;
pub use einsum::matmul;
pub use einsum::outer;
pub use einsum::trace;
pub use error::TensorError;
pub use tensor::Tensor;

Modules§

einsum
Einstein summation via TTGT (Transpose-Transpose-GEMM-Transpose).
error
Tensor error types.
tensor
Dense N-dimensional tensor.