qudit-core 0.3.3

Accelerated and Extensible Quantum Library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

//! Implements the tensor struct and associated methods for the Openqudit library.

mod symsq;
mod tensor;

pub use tensor::Tensor;
pub use tensor::TensorMut;
pub use tensor::TensorRef;

pub use symsq::SymSqTensor;
pub use symsq::SymSqTensorMut;
pub use symsq::SymSqTensorRef;

/// Helper for bounds checking
#[allow(dead_code)]
fn is_in_bounds<const D: usize>(indices: &[usize; D], dimensions: &[usize; D]) -> bool {
    for i in 0..D {
        if indices[i] >= dimensions[i] {
            return false;
        }
    }
    true
}

/// Helper for panic on out of bounds
fn check_bounds<const D: usize>(indices: &[usize; D], dimensions: &[usize; D]) {
    for i in 0..D {
        if indices[i] >= dimensions[i] {
            panic!(
                "Index out of bounds: index {} is {} but dimension {} has size {}",
                i, indices[i], i, dimensions[i]
            );
        }
    }
}

/// Calculates the strides for a tensor with the given dims in a contiguous, row-major, block.
fn calc_continuous_strides<const D: usize>(dims: &[usize; D]) -> [usize; D] {
    if D == 0 {
        [0; D]
    } else if D == 1 {
        [1; D]
    } else {
        let mut stride_acm = 1;
        let mut strides = [0; D];
        for (i, d) in dims.iter().enumerate().rev() {
            strides[i] = stride_acm;
            stride_acm *= d;
        }
        strides
    }
}