我正在Rust中编写一个并行矩阵乘法代码,我想并行计算乘积的每个元素。我使用ndarray
s来存储我的数据。因此,我的代码将是一些单独的行
fn mul(lhs: &Array2<f32>, rhs: &Array2<f32>) -> Array2<f32> {
let N = lhs.raw_size()[0];
let M = rhs.raw_size()[1];
let mut result = Array2::zeros((N,M));
range_2d(0..N,0..M).par_iter().map(|(i, j)| {
// load the result for the (i,j) element into 'result'
}).count();
result
}
字符串
有没有办法做到这一点?
1条答案
按热度按时间70gysomp1#
你可以用这种方式创建一个并行迭代器:
字符串