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
|
use ndarray::Array;
use ndarray::{arr1, arr2, Axis};
use num_complex::Complex;
use num_traits::Num;
fn c<T: Clone + Num>(re: T, im: T) -> Complex<T>
{
Complex::new(re, im)
}
#[test]
fn complex_mat_mul()
{
let a = arr2(&[[c(3., 4.), c(2., 0.)], [c(0., -2.), c(3., 0.)]]);
let b = (&a * c(3., 0.)).map(|c| 5. * c / c.norm_sqr());
println!("{:>8.2}", b);
let e = Array::eye(2);
let r = a.dot(&e);
println!("{}", a);
assert_eq!(r, a);
assert_eq!(
a.mean_axis(Axis(0)).unwrap(),
arr1(&[c(1.5, 1.), c(2.5, 0.)])
);
}
|
