spenso 0.5.3

A tensor (n-dim array) network, iterating, and contraction (using automatic abstract index matching) 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use std::{iter::Sum, ops::AddAssign};

use gat_lending_iterator::LendingIterator;

use crate::{
    structure::TensorStructure,
    tensors::data::{DataTensor, DenseTensor},
};

pub mod add;
pub mod add_assign;
pub mod algebraic_traits;
pub mod complex;
pub mod fallible_add;
pub mod fallible_sub;
pub mod neg;
pub mod ref_zero;
pub mod scalar_mul;
pub mod sub;
pub mod sub_assign;
pub mod upgrading_arithmetic;

impl<T, U, I> AddAssign<&DenseTensor<T, I>> for DenseTensor<U, I>
where
    U: for<'a> AddAssign<&'a T>,
    I: TensorStructure + Clone,
{
    fn add_assign(&mut self, rhs: &DenseTensor<T, I>) {
        for (u, t) in self.data.iter_mut().zip(rhs.data.iter()) {
            *u += t;
        }
    }
}

impl<T, S: TensorStructure + Clone> Sum for DataTensor<T, S>
where
    T: for<'a> AddAssign<&'a T>,
{
    fn sum<I: Iterator<Item = Self>>(mut iter: I) -> Self {
        if let Some(mut i) = iter.next() {
            for j in iter {
                i += j;
            }
            i
        } else {
            panic!("Empty iterator in sum");
        }
    }
}

// pub trait LendingSum: LendingIterator
// where
//     for<'p> Self::Item<'p>: Clone + for<'a> AddAssign<Self::Item>,
// {
//     fn sum_ref<I>(mut iter: I) -> Self {
//         if let Some(mut i) = iter.next() {
//             let sum = i.clone();

//             while let Some(j) = iter.next() {
//                 sum += j;
//             }
//             sum
//         } else {
//             panic!("Empty iterator in sum");
//         }
//     }
// }

impl<T, S: TensorStructure + Clone> DataTensor<T, S>
where
    T: for<'a> AddAssign<&'a T> + Clone,
{
    pub fn sum_ref<I>(mut iter: I) -> Self
    where
        for<'a> I: LendingIterator<Item<'a> = &'a DataTensor<T, S>>,
    {
        if let Some(i) = iter.next() {
            let mut sum = i.clone();

            while let Some(j) = iter.next() {
                sum += j;
            }
            sum
        } else {
            panic!("Empty iterator in sum");
        }
    }

    pub fn sum<I>(mut iter: I) -> Self
    where
        for<'a> I: LendingIterator<Item<'a> = DataTensor<T, S>>,
    {
        if let Some(i) = iter.next() {
            let mut sum = i;

            while let Some(j) = iter.next() {
                sum += j;
            }
            sum
        } else {
            panic!("Empty iterator in sum");
        }
    }
}

pub trait ScalarMul<T> {
    type Output;
    fn scalar_mul(&self, rhs: &T) -> Option<Self::Output>;
}

pub trait ScalarMulMut<T> {
    type Output;
    fn scalar_mul_mut(&self, rhs: &T) -> Option<Self::Output>;
}