acme-tensor 0.3.1

A comprehensive tensor library for Rust with support for automatic-differentiation.
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

/*
    Appellation: iter <impls>
    Contrib: FL03 <jo3mccain@icloud.com>
*/
use crate::shape::Axis;
use crate::tensor::TensorBase;
use core::iter::{Product, Sum};

impl<T> TensorBase<T>
where
    T: Copy,
{
    /// Compute the product of all elements in the tensor
    pub fn product(&self) -> T
    where
        T: Product,
    {
        self.data().iter().copied().product()
    }
    #[doc(hidden)]
    pub fn product_axis(&self, _axis: Axis) -> T {
        unimplemented!("product_axis")
    }
    /// Compute the sum of all elements in the tensor
    pub fn sum(&self) -> T
    where
        T: Sum,
    {
        self.data().iter().copied().sum()
    }
    #[doc(hidden)]
    /// Compute the sum of all elements along the given axis
    pub fn sum_axis(&self, _axis: Axis) -> T {
        unimplemented!("sum_axis")
    }
}

impl<T> IntoIterator for TensorBase<T> {
    type Item = T;
    type IntoIter = std::vec::IntoIter<T>;

    fn into_iter(self) -> Self::IntoIter {
        self.data.into_iter()
    }
}

impl<T> FromIterator<T> for TensorBase<T> {
    fn from_iter<I: IntoIterator<Item = T>>(iter: I) -> Self {
        Self::from_vec(Vec::from_iter(iter))
    }
}