ndarray-linalg 0.18.1

Linear algebra package for rust-ndarray using LAPACK
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

use crate::types::*;
use ndarray::*;

/// Inner Product
///
/// Differenct from `Dot` trait, this take complex conjugate of `self` elements
///
pub trait InnerProduct {
    type Elem: Scalar;

    /// Inner product `(self.conjugate, rhs)
    fn inner(&self, rhs: &ArrayRef<Self::Elem, Ix1>) -> Self::Elem;
}

impl<A> InnerProduct for ArrayRef<A, Ix1>
where
    A: Scalar,
{
    type Elem = A;

    fn inner(&self, rhs: &ArrayRef<A, Ix1>) -> A {
        assert_eq!(self.len(), rhs.len());
        Zip::from(self)
            .and(rhs)
            .fold_while(A::zero(), |acc, s, r| {
                FoldWhile::Continue(acc + s.conj() * *r)
            })
            .into_inner()
    }
}