[ndarray] Add `NDAxisIter` struct and `iter_by_axis` method that iterate array by any axis#212

shivasankarka merged 3 commits intoMojo-Numerics-and-Algorithms-group:pre-0.6Mojo-Numerics-and-Algorithms-group/NuMojo:pre-0.6from

forfudan:iterforfudan/NuMojo:iterCopy head branch name to clipboard