trueno_tensor/lib.rs
1#![cfg_attr(
2 test,
3 allow(
4 clippy::expect_used,
5 clippy::unwrap_used,
6 clippy::disallowed_methods,
7 clippy::float_cmp,
8 clippy::panic
9 )
10)]
11//! N-dimensional tensor contractions — Einstein summation via TTGT.
12//!
13//! This crate provides a dense tensor type and Einstein summation (`einsum`)
14//! that reduces tensor contractions to explicit loops, following the TTGT
15//! (Transpose-Transpose-GEMM-Transpose) strategy for cuTENSOR parity.
16
17pub mod einsum;
18pub mod error;
19pub mod tensor;
20
21pub use einsum::{batch_matmul, einsum, einsum_nary, matmul, outer, trace};
22pub use error::TensorError;
23pub use tensor::Tensor;
24
25#[cfg(test)]
26mod tests;