METHOD_0 ( ) { double [ ] [ ] [ ] IDENT_0 = new double [ 3 ] [ ] [ ] ; boolean [ ] [ ] [ ] IDENT_1 = new boolean [ 3 ] [ ] [ ] ; IDENT_0 [ 0 ] = new double [ ] [ ] { new double [ ] { 1 , 10 , 2 } , new double [ ] { 3 , 4 , 5 } } ; IDENT_0 [ 1 ] = new double [ ] [ ] { new double [ ] { - 10 , - 9 , - 8 } , new double [ ] { - 7 , - 6 , - 5 } } ; IDENT_0 [ 2 ] = new double [ ] [ ] { new double [ ] { 4 , 3 , 2 } , new double [ ] { 1 , 0 , - 1 } } ; IDENT_1 [ 0 ] = new boolean [ ] [ ] { new boolean [ ] { false , true , false } , new boolean [ ] { false , false , false } } ; IDENT_1 [ 1 ] = new boolean [ ] [ ] { new boolean [ ] { false , false , false } , new boolean [ ] { false , false , true } } ; IDENT_1 [ 2 ] = new boolean [ ] [ ] { new boolean [ ] { true , false , false } , new boolean [ ] { false , false , false } } ; org . nd4j . linalg . api . ndarray . INDArray arr = org . nd4j . linalg . factory . Nd4j . create ( 3 , 2 , 3 ) ; org . nd4j . linalg . api . ndarray . INDArray expected = org . nd4j . linalg . factory . Nd4j . create ( IDENT_2 . IDENT_3 , 3 , 2 , 3 ) ; for ( int i = 0 ; i < 3 ; i ++ ) { arr . get ( org . nd4j . linalg . indexing . IDENT_4 . METHOD_1 ( i ) , org . nd4j . linalg . indexing . IDENT_4 . METHOD_2 ( ) , org . nd4j . linalg . indexing . IDENT_4 . METHOD_2 ( ) ) . METHOD_3 ( org . nd4j . linalg . factory . Nd4j . create ( IDENT_0 [ i ] ) ) ; IDENT_5 . val t = org . nd4j . linalg . factory . Nd4j . create ( org . nd4j . linalg . util . IDENT_6 . METHOD_4 ( IDENT_1 [ i ] ) , new long [ ] { 2 , 3 } , IDENT_2 . IDENT_3 ) ; IDENT_5 . val v = expected . get ( org . nd4j . linalg . indexing . IDENT_4 . METHOD_1 ( i ) , org . nd4j . linalg . indexing . IDENT_4 . METHOD_2 ( ) , org . nd4j . linalg . indexing . IDENT_4 . METHOD_2 ( ) ) ; v . METHOD_3 ( t ) ; } IDENT_5 . val result = org . nd4j . linalg . factory . Nd4j . METHOD_5 ( ) . exec ( new org . nd4j . linalg . api . IDENT_7 . impl . IDENT_8 . any . METHOD_6 ( arr , org . nd4j . linalg . factory . Nd4j . METHOD_7 ( IDENT_2 . IDENT_3 , arr . METHOD_8 ( ) , arr . METHOD_9 ( ) ) , 1 , 2 ) ) ; "<AssertPlaceHolder>" ; } METHOD_9 ( ) { return STRING_0 ; }
org . junit . Assert . assertEquals ( expected , result )