qudit-tensor 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

#![allow(dead_code)]

use qudit_core::ComplexScalar;
use qudit_core::memory::MemoryBuffer;
use qudit_expr::{DifferentiationLevel, FUNCTION, GRADIENT, HESSIAN};

use super::super::buffer::SizedTensorBuffer;

pub struct TraceStruct<C: ComplexScalar> {
    pub input: SizedTensorBuffer<C>,
    pub output: SizedTensorBuffer<C>,
}

impl<C: ComplexScalar> TraceStruct<C> {
    pub fn new(
        input: SizedTensorBuffer<C>,
        _dim_pairs: Vec<(usize, usize)>,
        output: SizedTensorBuffer<C>,
        diff_lvl: DifferentiationLevel,
    ) -> Self {
        assert!(input.nparams() == output.nparams());

        // Trace instruction is extremely likely, if always possible, to
        // optimize out. Leaving as a todo for now to focus on more important
        // things.
        if diff_lvl == FUNCTION {
            todo!()
        } else if diff_lvl == GRADIENT {
            todo!()
        } else if diff_lvl == HESSIAN {
            todo!()
        } else {
            panic!("Invalid differentiation level.");
        };

        // Self {
        //     input,
        //     output,
        // }
    }

    #[inline(always)]
    pub unsafe fn evaluate(&self, _memory: &mut MemoryBuffer<C>) {
        todo!()
    }
}