@@ -222,5 +222,19 @@ fn test_reshape() {
222
222
// broadcast shapes
223
223
test_reshape ! ( ok C from [ 3 , 4 , 5 , 7 ] , [ 0 , 0 , 7 , 1 ] , to [ 12 , 35 ] , [ 0 , 1 ] ) ;
224
224
test_reshape ! ( fail C from [ 3 , 4 , 5 , 7 ] , [ 0 , 0 , 7 , 1 ] , to [ 28 , 15 ] ) ;
225
+
226
+ // one-filled shapes
227
+ test_reshape ! ( ok C from [ 10 ] , [ 1 ] , to [ 1 , 10 , 1 , 1 , 1 ] , [ 1 , 1 , 1 , 1 , 1 ] ) ;
228
+ test_reshape ! ( ok F from [ 10 ] , [ 1 ] , to [ 1 , 10 , 1 , 1 , 1 ] , [ 1 , 1 , 1 , 1 , 1 ] ) ;
229
+ test_reshape ! ( ok C from [ 1 , 10 ] , [ 10 , 1 ] , to [ 1 , 10 , 1 , 1 , 1 ] , [ 10 , 1 , 1 , 1 , 1 ] ) ;
230
+ test_reshape ! ( ok F from [ 1 , 10 ] , [ 10 , 1 ] , to [ 1 , 10 , 1 , 1 , 1 ] , [ 10 , 1 , 1 , 1 , 1 ] ) ;
231
+ test_reshape ! ( ok C from [ 1 , 10 ] , [ 1 , 1 ] , to [ 1 , 5 , 1 , 1 , 2 ] , [ 1 , 2 , 2 , 2 , 1 ] ) ;
232
+ test_reshape ! ( ok F from [ 1 , 10 ] , [ 1 , 1 ] , to [ 1 , 5 , 1 , 1 , 2 ] , [ 1 , 1 , 5 , 5 , 5 ] ) ;
233
+ test_reshape ! ( ok C from [ 10 , 1 , 1 , 1 , 1 ] , [ 1 , 1 , 1 , 1 , 1 ] , to [ 10 ] , [ 1 ] ) ;
234
+ test_reshape ! ( ok F from [ 10 , 1 , 1 , 1 , 1 ] , [ 1 , 1 , 1 , 1 , 1 ] , to [ 10 ] , [ 1 ] ) ;
235
+ test_reshape ! ( ok C from [ 1 , 5 , 1 , 2 , 1 ] , [ 1 , 2 , 1 , 1 , 1 ] , to [ 10 ] , [ 1 ] ) ;
236
+ test_reshape ! ( fail F from [ 1 , 5 , 1 , 2 , 1 ] , [ 1 , 2 , 1 , 1 , 1 ] , to [ 10 ] ) ;
237
+ test_reshape ! ( ok F from [ 1 , 5 , 1 , 2 , 1 ] , [ 1 , 1 , 1 , 5 , 1 ] , to [ 10 ] , [ 1 ] ) ;
238
+ test_reshape ! ( fail C from [ 1 , 5 , 1 , 2 , 1 ] , [ 1 , 1 , 1 , 5 , 1 ] , to [ 10 ] ) ;
225
239
}
226
240
0 commit comments