并行枚举 rust ndarray 数组的正确方法

Question

并行枚举 rust ndarray 数组的正确方法是什么？

目前我使用：

depth.indexed_iter_mut().par_bridge()

我相信有一个API可以做到这一点，但我只能找到

par_iter

和

par_iter_mut

，我不能在其上调用

.enumerate()

。

已经浏览过文档和教程，搜索过google等

Answer 1

Enumerate 返回

usize

索引，但

ndarray::Array

可能具有多个维度。你需要选择一个

Axis

来枚举，比如

Axis(0)

:

use ndarray::{array, Array2, Axis};

fn main() {
    let mut arr: Array2<f64> = array![[1., 2.], [3., 4.], [5., 6.]];
    arr.axis_iter_mut(Axis(0)).enumerate().for_each(|(i, a)| {
        println!("{a:?}");
    });
}