@@ -5,12 +5,12 @@ use ndarray::prelude::*;
5
5
fn broadcast_1 ( ) {
6
6
let a_dim = Dim ( [ 2 , 4 , 2 , 2 ] ) ;
7
7
let b_dim = Dim ( [ 2 , 1 , 2 , 1 ] ) ;
8
- let a = ArcArray :: linspace ( 0. , 1. , a_dim. size ( ) ) . reshape ( a_dim) ;
9
- let b = ArcArray :: linspace ( 0. , 1. , b_dim. size ( ) ) . reshape ( b_dim) ;
8
+ let a = ArcArray :: linspace ( 0. , 1. , a_dim. size ( ) ) . into_shape ( a_dim) . unwrap ( ) ;
9
+ let b = ArcArray :: linspace ( 0. , 1. , b_dim. size ( ) ) . into_shape ( b_dim) . unwrap ( ) ;
10
10
assert ! ( b. broadcast( a. dim( ) ) . is_some( ) ) ;
11
11
12
12
let c_dim = Dim ( [ 2 , 1 ] ) ;
13
- let c = ArcArray :: linspace ( 0. , 1. , c_dim. size ( ) ) . reshape ( c_dim) ;
13
+ let c = ArcArray :: linspace ( 0. , 1. , c_dim. size ( ) ) . into_shape ( c_dim) . unwrap ( ) ;
14
14
assert ! ( c. broadcast( 1 ) . is_none( ) ) ;
15
15
assert ! ( c. broadcast( ( ) ) . is_none( ) ) ;
16
16
assert ! ( c. broadcast( ( 2 , 1 ) ) . is_some( ) ) ;
@@ -31,8 +31,8 @@ fn broadcast_1() {
31
31
fn test_add ( ) {
32
32
let a_dim = Dim ( [ 2 , 4 , 2 , 2 ] ) ;
33
33
let b_dim = Dim ( [ 2 , 1 , 2 , 1 ] ) ;
34
- let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . reshape ( a_dim) ;
35
- let b = ArcArray :: linspace ( 0.0 , 1. , b_dim. size ( ) ) . reshape ( b_dim) ;
34
+ let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . into_shape ( a_dim) . unwrap ( ) ;
35
+ let b = ArcArray :: linspace ( 0.0 , 1. , b_dim. size ( ) ) . into_shape ( b_dim) . unwrap ( ) ;
36
36
a += & b;
37
37
let t = ArcArray :: from_elem ( ( ) , 1.0f32 ) ;
38
38
a += & t;
@@ -43,7 +43,7 @@ fn test_add() {
43
43
#[ cfg( feature = "std" ) ]
44
44
fn test_add_incompat ( ) {
45
45
let a_dim = Dim ( [ 2 , 4 , 2 , 2 ] ) ;
46
- let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . reshape ( a_dim) ;
46
+ let mut a = ArcArray :: linspace ( 0.0 , 1. , a_dim. size ( ) ) . into_shape ( a_dim) . unwrap ( ) ;
47
47
let incompat = ArcArray :: from_elem ( 3 , 1.0f32 ) ;
48
48
a += & incompat;
49
49
}
0 commit comments