如何使用 tokio、async-std 并行化几分钟

问题描述

我想使用 tokio 或 async-std 在多线程中执行以下处理。 我已经阅读了关于 tokio 和 async-std 的教程，但我认为没有提到并行化 for。 在这个过程中，所有线程都引用同一个数组索引。此外，所有线程将访问同一数组的不同位置。

let input_array: Array2<f32>;  
let output_array: Array2<f32>;
for i in 0..roop_num {
  let res = do_some_func(&input_array,idx);
  output_array.slice_mut(s![idx,...]) .assign(&res);
}

我想把上面代码的for部分改成并行处理。

解决方法

Tokio 或 async-std 处理并发，而不是并行。如果您需要数据并行，那么 rayon 是一个可供选择的库。如果您使用的是 Iterator trait，那么 chunks() 方法很好。对于更命令式的方法，您可以使用 par_chunks_mut