@@ -462,10 +462,9 @@ macro_rules! arith_assign_func {
462462 fn $fn_name( & mut self , rhs: Array ) {
463463 let tmp_seq = Seq :: <f32 >:: default ( ) ;
464464 let mut idxrs = Indexer :: new( ) ;
465- idxrs. set_index( & tmp_seq, 0 , Some ( false ) ) ;
466- idxrs. set_index( & tmp_seq, 1 , Some ( false ) ) ;
467- idxrs. set_index( & tmp_seq, 2 , Some ( false ) ) ;
468- idxrs. set_index( & tmp_seq, 3 , Some ( false ) ) ;
465+ for n in 0 ..self . numdims( ) {
466+ idxrs. set_index( & tmp_seq, n, Some ( false ) ) ;
467+ }
469468 let tmp = assign_gen( self as & Array , & idxrs,
470469 & $func( self as & Array , & rhs, false ) ) ;
471470 let old = mem:: replace( self , tmp) ;
@@ -490,10 +489,9 @@ macro_rules! bit_assign_func {
490489 fn $fn_name( & mut self , rhs: Array ) {
491490 let tmp_seq = Seq :: <f32 >:: default ( ) ;
492491 let mut idxrs = Indexer :: new( ) ;
493- idxrs. set_index( & tmp_seq, 0 , Some ( false ) ) ;
494- idxrs. set_index( & tmp_seq, 1 , Some ( false ) ) ;
495- idxrs. set_index( & tmp_seq, 2 , Some ( false ) ) ;
496- idxrs. set_index( & tmp_seq, 3 , Some ( false ) ) ;
492+ for n in 0 ..self . numdims( ) {
493+ idxrs. set_index( & tmp_seq, n, Some ( false ) ) ;
494+ }
497495 let tmp = assign_gen( self as & Array , & idxrs, & $func( self as & Array , & rhs, false ) ) ;
498496 let old = mem:: replace( self , tmp) ;
499497 }
0 commit comments