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

问题描述 投票:0回答:1

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

目前我使用:

depth.indexed_iter_mut().par_bridge()

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

par_iter
par_iter_mut
,我不能在其上调用
.enumerate()

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

rust multidimensional-array linear-algebra
1个回答
0
投票

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:?}");
    });
}
最新问题
© www.soinside.com 2019 - 2024. All rights reserved.