@@ -13,14 +13,22 @@ fn cholesky() {
13
13
println!( "a = \n {:?}" , a_orig) ;
14
14
15
15
let upper = a_orig. cholesky( UPLO :: Upper ) . unwrap( ) ;
16
- assert_close_l2!( & upper. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) , & a_orig, $rtol) ;
16
+ assert_close_l2!(
17
+ & upper. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) ,
18
+ & a_orig,
19
+ $rtol
20
+ ) ;
17
21
18
22
let lower = a_orig. cholesky( UPLO :: Lower ) . unwrap( ) ;
19
23
assert_close_l2!( & lower. dot( & lower. t( ) . mapv( |elem| elem. conj( ) ) ) , & a_orig, $rtol) ;
20
24
21
25
let a: Array2 <$elem> = replicate( & a_orig) ;
22
26
let upper = a. cholesky_into( UPLO :: Upper ) . unwrap( ) ;
23
- assert_close_l2!( & upper. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) , & a_orig, $rtol) ;
27
+ assert_close_l2!(
28
+ & upper. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) ,
29
+ & a_orig,
30
+ $rtol
31
+ ) ;
24
32
25
33
let a: Array2 <$elem> = replicate( & a_orig) ;
26
34
let lower = a. cholesky_into( UPLO :: Lower ) . unwrap( ) ;
@@ -29,7 +37,11 @@ fn cholesky() {
29
37
let mut a: Array2 <$elem> = replicate( & a_orig) ;
30
38
{
31
39
let upper = a. cholesky_inplace( UPLO :: Upper ) . unwrap( ) ;
32
- assert_close_l2!( & upper. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) , & a_orig, $rtol) ;
40
+ assert_close_l2!(
41
+ & upper. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) ,
42
+ & a_orig,
43
+ $rtol
44
+ ) ;
33
45
}
34
46
assert_close_l2!( & a. t( ) . mapv( |elem| elem. conj( ) ) . dot( & upper. view( ) ) , & a_orig, $rtol) ;
35
47
@@ -103,7 +115,11 @@ fn cholesky_det() {
103
115
( $elem: ty, $atol: expr) => {
104
116
let a: Array2 <$elem> = random_hpd( 3 ) ;
105
117
println!( "a = \n {:?}" , a) ;
106
- let ln_det = a. eigvalsh( UPLO :: Upper ) . unwrap( ) . mapv( |elem| elem. ln( ) ) . scalar_sum( ) ;
118
+ let ln_det = a
119
+ . eigvalsh( UPLO :: Upper )
120
+ . unwrap( )
121
+ . mapv( |elem| elem. ln( ) )
122
+ . scalar_sum( ) ;
107
123
let det = ln_det. exp( ) ;
108
124
assert_aclose!( a. factorizec( UPLO :: Upper ) . unwrap( ) . detc( ) , det, $atol) ;
109
125
assert_aclose!( a. factorizec( UPLO :: Upper ) . unwrap( ) . ln_detc( ) , ln_det, $atol) ;
0 commit comments