rust - 在 Rust 中计算两个 f64 向量的点积的最快方法是什么？

Question

我正在用 Rust 编写神经网络的实现，并尝试计算两个矩阵的点积。我有以下代码:

fn dot_product(a: Vec<f64>, b: Vec<f64>) -> f64 {
    // Calculate the dot product of two vectors.
    assert_eq!(a.len(), b.len());
    let mut product: f64 = 0.0;
    for i in 0..a.len() {
        product += a[i] * b[i];
    }
    product
}

这需要两个向量，a 和 b(长度相同)并执行逐元素乘法(乘以向量 a 的值 1 code> 具有向量 b 的值 1，并将其添加到向量 a 的值 2 和向量 b 的值 2 等等。 .).

有没有更有效的方法，如果有，怎么做？

Answer 1

这并不是一个全面的一般性答案，但我想分享一些代码。

除非我知道这是我的应用程序的瓶颈，否则您的实现看起来很像我会做的。然后我会研究更深奥的方法(可能是 SIMD )。

也就是说，您可能会考虑更改您的函数以改为采用切片引用。这样你就可以传递 Vec 或数组:

fn dot_product(a: &[f64], b: &[f64]) -> f64 {
    // Calculate the dot product of two vectors. 
    assert_eq!(a.len(), b.len()); 
    let mut product = 0.0;
    for i in 0..a.len() {
        product += a[i] * b[i];
    }
    product
}

fn main() {
    println!("{}", dot_product(&[1.0,2.0], &[3.0,4.0]));
    println!("{}", dot_product(&vec![1.0,2.0], &vec![3.0,4.0]));
}

另见:

• Why is it discouraged to accept a reference to a String (&String), Vec (&Vec), or Box (&Box) as a function argument?