burn_quantum 0.3.0

Differentiable qubit and qudit simulation with Burn's native tensor and module APIs
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

//! Trace-preserving attenuation into vacuum, including asymmetric rail loss.
use crate::{Error, KrausChannel, Matrix, Result};
use serde::{Deserialize, Serialize};

/// Independent intensity transmission probabilities for the logical-zero/one rails.
#[derive(Clone, Copy, Debug, PartialEq, Serialize, Deserialize)]
#[serde(try_from = "[f64; 2]", into = "[f64; 2]")]
pub struct RailTransmission([f64; 2]);
impl RailTransmission {
    /// Perfect transmission on both rails.
    pub const IDEAL: Self = Self([1., 1.]);
    /// Validate intensity (not amplitude) transmissions in `[0,1]`.
    pub fn new(zero: f64, one: f64) -> Result<Self> {
        if [zero, one]
            .iter()
            .any(|p| !p.is_finite() || !(0.0..=1.0).contains(p))
        {
            return Err(Error::InvalidValue(
                "rail transmission must be finite and in [0,1]".into(),
            ));
        }
        Ok(Self([zero, one]))
    }
    /// Equal transmission on both rails.
    pub fn uniform(value: f64) -> Result<Self> {
        Self::new(value, value)
    }
    /// Ordered rail transmission probabilities.
    pub fn values(self) -> [f64; 2] {
        self.0
    }
    /// Compose independent attenuation along consecutive path segments.
    pub fn then(self, next: Self) -> Self {
        Self([self.0[0] * next.0[0], self.0[1] * next.0[1]])
    }
    /// Apply a per-tick storage transmission for an integer number of ticks.
    pub fn over_ticks(self, mut ticks: u64) -> Self {
        let mut out = Self::IDEAL;
        let mut factor = self;
        while ticks > 0 {
            if ticks & 1 == 1 {
                out = out.then(factor);
            }
            factor = factor.then(factor);
            ticks >>= 1;
        }
        out
    }
    /// CPTP channel with separate environment records for loss from each rail.
    ///
    /// K0=diag(sqrt(eta0),sqrt(eta1),1), K1=sqrt(1-eta0)|vac><0|,
    /// K2=sqrt(1-eta1)|vac><1|. Vacuum is absorbing. Loss parameters are
    /// constant; gradients through an input density matrix are retained.
    pub fn channel(self) -> Result<KrausChannel> {
        let [a, b] = self.0;
        KrausChannel::new(vec![
            Matrix::real(3, vec![a.sqrt(), 0., 0., 0., b.sqrt(), 0., 0., 0., 1.])?,
            Matrix::real(3, vec![0., 0., 0., 0., 0., 0., (1. - a).sqrt(), 0., 0.])?,
            Matrix::real(3, vec![0., 0., 0., 0., 0., 0., 0., (1. - b).sqrt(), 0.])?,
        ])
    }
}
impl TryFrom<[f64; 2]> for RailTransmission {
    type Error = Error;
    fn try_from(v: [f64; 2]) -> Result<Self> {
        Self::new(v[0], v[1])
    }
}
impl From<RailTransmission> for [f64; 2] {
    fn from(v: RailTransmission) -> Self {
        v.0
    }
}