ndarray_linalg/
diagonal.rs

1//! Vector as a Diagonal matrix
2
3use ndarray::*;
4
5use super::operator::*;
6use super::types::*;
7
8/// Vector as a Diagonal matrix
9pub struct Diagonal<S: Data> {
10    diag: ArrayBase<S, Ix1>,
11}
12
13pub trait IntoDiagonal<S: Data> {
14    fn into_diagonal(self) -> Diagonal<S>;
15}
16
17pub trait AsDiagonal<A> {
18    fn as_diagonal(&self) -> Diagonal<ViewRepr<&A>>;
19}
20
21impl<S: Data> IntoDiagonal<S> for ArrayBase<S, Ix1> {
22    fn into_diagonal(self) -> Diagonal<S> {
23        Diagonal { diag: self }
24    }
25}
26
27impl<A> AsDiagonal<A> for ArrayRef<A, Ix1> {
28    fn as_diagonal(&self) -> Diagonal<ViewRepr<&A>> {
29        Diagonal { diag: self.view() }
30    }
31}
32
33impl<A, Sa> LinearOperator for Diagonal<Sa>
34where
35    A: Scalar,
36    Sa: Data<Elem = A>,
37{
38    type Elem = A;
39
40    fn apply_mut(&self, a: &mut ArrayRef<A, Ix1>) {
41        for (val, d) in a.iter_mut().zip(self.diag.iter()) {
42            *val *= *d;
43        }
44    }
45}