tensor4all_core/
lib.rs

1//! Core tensor operations and types for tensor4all-rs.
2//!
3//! This crate provides the foundational types and operations for tensor networks:
4//!
5//! - **Index types**: [`DynIndex`], [`Index`], [`DynId`] for tensor indices
6//! - **Tag sets**: [`TagSet`], [`TagSetLike`] for metadata tagging
7//! - **Tensors**: [`TensorDynLen`] for dynamic-rank dense tensors
8//! - **Operations**: Contraction, SVD, QR decomposition, factorization
9//!
10//! # Example
11//!
12//! ```
13//! use tensor4all_core::{Index, DynIndex, TensorDynLen};
14//!
15//! // Create indices with dynamic identity
16//! let i = Index::new_dyn(2);
17//! let j = Index::new_dyn(3);
18//!
19//! // Create a tensor
20//! let data = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0];
21//! let t = TensorDynLen::from_dense(vec![i.clone(), j.clone()], data).unwrap();
22//! ```
23
24#![warn(missing_docs)]
25
26pub mod col_major_array;
27pub use col_major_array::{ColMajorArray, ColMajorArrayMut, ColMajorArrayRef};
28
29// Common (tags, utilities, scalar)
30/// Dynamic scalar compatibility wrapper built on rank-0 `TensorDynLen`.
31pub mod any_scalar;
32pub mod global_default;
33pub mod index_like;
34pub mod scalar;
35/// Stack-allocated fixed-capacity string types for ITensors.jl compatibility.
36pub mod smallstring;
37/// Tag set types for tensor metadata.
38pub mod tagset;
39pub mod truncation;
40
41pub use scalar::CommonScalar;
42
43// Default concrete type implementations (index, tensor, linalg, etc.)
44pub mod defaults;
45
46// Backwards compatibility: re-export defaults submodules as top-level modules
47// This allows `tensor4all_core::index::...` to work
48pub use defaults::index;
49
50pub use defaults::{DefaultIndex, DefaultTagSet, DynId, DynIndex, Index, TagSet};
51pub use index_like::{ConjState, IndexLike};
52
53/// Index operations (replacement, set operations, contraction preparation).
54pub mod index_ops;
55pub use index_ops::{
56    check_unique_indices, common_ind_positions, common_inds, hascommoninds, hasind, hasinds,
57    noncommon_inds, prepare_contraction, prepare_contraction_pairs, replaceinds,
58    replaceinds_in_place, union_inds, unique_inds, ContractionError, ContractionSpec,
59    ReplaceIndsError,
60};
61pub use smallstring::{SmallChar, SmallString, SmallStringError};
62pub use tagset::{Tag, TagSetError, TagSetLike};
63
64// Tensor (storage, tensor types)
65pub mod tensor_index;
66pub mod tensor_like;
67
68pub use tensor_index::TensorIndex;
69
70// Krylov subspace methods (GMRES, etc.)
71pub mod krylov;
72
73// Block tensor for block matrix GMRES
74pub mod block_tensor;
75
76// Backwards compatibility: re-export defaults::tensordynlen as tensor
77pub use defaults::tensordynlen as tensor;
78
79pub use any_scalar::AnyScalar;
80pub use defaults::tensordynlen::{
81    compute_permutation_from_indices, diag_tensor_dyn_len, unfold_split, TensorDynLen,
82};
83pub use tensor4all_tensorbackend::TensorElement;
84pub use tensor4all_tensorbackend::{
85    print_and_reset_native_einsum_profile, reset_native_einsum_profile,
86};
87pub use tensor_like::{
88    AllowedPairs, Canonical, DirectSumResult, FactorizeAlg, FactorizeError, FactorizeOptions,
89    FactorizeResult, LinearizationOrder, TensorLike,
90};
91
92// Contraction - backwards compatibility
93pub use defaults::contract;
94pub use defaults::contract::{
95    contract_connected, contract_connected_with_options, contract_multi, contract_multi_owned,
96    contract_multi_with_options, print_and_reset_contract_profile, reset_contract_profile,
97    ContractionOptions,
98};
99
100// Re-export linear algebra modules from defaults for backwards compatibility
101// This allows `tensor4all_core::svd::...`, `tensor4all_core::qr::...`, etc.
102pub mod direct_sum {
103    //! Re-export of direct sum operations.
104    pub use crate::defaults::direct_sum::*;
105}
106pub mod factorize {
107    //! Re-export of factorization operations.
108    pub use crate::defaults::factorize::*;
109}
110pub mod qr {
111    //! Re-export of QR decomposition operations.
112    pub use crate::defaults::qr::*;
113}
114pub mod svd {
115    //! Re-export of SVD decomposition operations.
116    pub use crate::defaults::svd::{
117        default_svd_truncation_policy, set_default_svd_truncation_policy, svd, svd_with, SvdError,
118        SvdOptions,
119    };
120}
121
122// Re-export linear algebra items for top-level access
123pub use defaults::direct_sum::direct_sum;
124pub use defaults::factorize::{factorize, factorize_full_rank};
125pub use defaults::qr::{default_qr_rtol, qr, qr_with, set_default_qr_rtol, QrError, QrOptions};
126pub use defaults::svd::{
127    default_svd_truncation_policy, set_default_svd_truncation_policy, svd, svd_with, SvdError,
128    SvdOptions,
129};
130
131// Global default and truncation utilities
132pub use global_default::{GlobalDefault, InvalidRtolError};
133pub use truncation::{
134    DecompositionAlg, InvalidThresholdError, SingularValueMeasure, SvdTruncationPolicy,
135    ThresholdScale, TruncationRule,
136};
tensor4all_core/lib.rs

tensor4all_core/
lib.rs
