mdarray-linalg 0.2.0

Linear algebra operations for mdarray, with multiple exchangeable backends
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

use mdarray::{Dim, Layout, Shape, Slice};
use num_complex::ComplexFloat;

/// Performs naively A += α·x·yᵀ
pub fn naive_outer<T: ComplexFloat, La: Layout, Lx: Layout, Ly: Layout, D0, D1>(
    a: &mut Slice<T, (D0, D1), La>,
    x: &Slice<T, (D0,), Lx>,
    y: &Slice<T, (D1,), Ly>,
    alpha: T,
) where
    D0: Dim,
    D1: Dim,
{
    let m = x.shape().dim(0);
    let n = y.shape().dim(0);

    let ash = *a.shape();
    assert!(
        ash.dim(0) == m,
        "Output shape must match input vector length"
    );
    assert!(
        ash.dim(1) == n,
        "Output shape must match input vector length"
    );

    for i in 0..m {
        for j in 0..n {
            a[[i, j]] = a[[i, j]] + alpha * x[[i]] * y[[j]];
        }
    }
}