<s> import arrayfire as af <EOL> from arrayfire import ParallelRange <EOL> import array as host <EOL> from . import _util <EOL> def simple_index ( verbose = False ) : <EOL> display_func = _util . display_func ( verbose ) <EOL> print_func = _util . print_func ( verbose ) <EOL> a = af . randu ( <NUM_LIT:5> , <NUM_LIT:5> ) <EOL> display_func ( a ) <EOL> b = af . Array ( a ) <EOL> display_func ( b ) <EOL> c = a . copy ( ) <EOL> display_func ( c ) <EOL> display_func ( a [ <NUM_LIT:0> , <NUM_LIT:0> ] ) <EOL> display_func ( a [ <NUM_LIT:0> ] ) <EOL> display_func ( a [ : ] ) <EOL> display_func ( a [ : , : ] ) <EOL> display_func ( a [ <NUM_LIT:0> : <NUM_LIT:3> , ] ) <EOL> display_func ( a [ - <NUM_LIT:2> : - <NUM_LIT:1> , - <NUM_LIT:1> ] ) <EOL> display_func ( a [ <NUM_LIT:0> : <NUM_LIT:5> ] ) <EOL> display_func ( a [ <NUM_LIT:0> : <NUM_LIT:5> : <NUM_LIT:2> ] ) <EOL> idx = af . Array ( host . array ( '<STR_LIT:i>' , [ <NUM_LIT:0> , <NUM_LIT:3> , <NUM_LIT:2> ] ) ) <EOL> display_func ( idx ) <EOL> aa = a [ idx ] <EOL> display_func ( aa ) <EOL> a [ <NUM_LIT:0> ] = <NUM_LIT:1> <EOL> display_func ( a ) <EOL> a [ <NUM_LIT:0> ] = af . randu ( <NUM_LIT:1> , <NUM_LIT:5> ) <EOL> display_func ( a ) <EOL> a [ : ] = af . randu ( <NUM_LIT:5> , <NUM_LIT:5> ) <EOL> display_func ( a ) <EOL> a [ : , - <NUM_LIT:1> ] = af . randu ( <NUM_LIT:5> , <NUM_LIT:1> ) <EOL> display_func ( a ) <EOL> a [ <NUM_LIT:0> : <NUM_LIT:5> : <NUM_LIT:2> ] = af . randu ( <NUM_LIT:3> , <NUM_LIT:5> ) <EOL> display_func ( a ) <EOL> a [ idx , idx ] = af . randu ( <NUM_LIT:3> , <NUM_LIT:3> ) <EOL> display_func ( a ) <EOL> a = af . randu ( <NUM_LIT:5> , <NUM_LIT:1> ) <EOL> b = af . randu ( <NUM_LIT:5> , <NUM_LIT:1> ) <EOL> display_func ( a ) <EOL> display_func ( b ) <EOL> for ii in ParallelRange ( <NUM_LIT:1> , <NUM_LIT:3> ) : <EOL> a [ ii ] = b [ ii ] <EOL> display_func ( a ) <EOL> for ii in ParallelRange ( <NUM_LIT:2> , <NUM_LIT:5> ) : <EOL> b [ ii ] = <NUM_LIT:2> <EOL> display_func ( b ) <EOL> a = af . randu ( <NUM_LIT:3> , <NUM_LIT:2> ) <EOL> rows = af . constant ( <NUM_LIT:0> , <NUM_LIT:1> , dtype = af . Dtype . s32 ) <EOL> b = a [ : , rows ] <EOL> display_func ( b ) <EOL> for r in rows : <EOL> display_func ( r ) <EOL> display_func ( b [ : , r ] ) <EOL> a = af . randu ( <NUM_LIT:3> ) <EOL> c = af . randu ( <NUM_LIT:3> ) <EOL> b = af . constant ( <NUM_LIT:1> , <NUM_LIT:3> , dtype = af . Dtype . b8 ) <EOL> display_func ( a ) <EOL> a [ b ] = c <EOL> display_func ( a ) <EOL> _util . tests [ '<STR_LIT:index>' ] = simple_index </s>
