struct - 方法 `mul` 具有不兼容的特征类型

Question

我正在用 Rust 创建一个简单的矩阵结构，我正在尝试实现一些基本的运算符方法:

use std::ops::Mul;

struct Matrix {
    cols: i32,
    rows: i32,
    data: Vec<f32>,
}

impl Matrix {
    fn new(cols: i32, rows: i32, data: Vec<f32>) -> Matrix {
        Matrix {
            cols: cols,
            rows: rows,
            data: data,
        }
    }
}

impl Mul<f32> for Matrix {
    type Output = Matrix;

    fn mul(&self, m: f32) -> Matrix {
        let mut new_data = Vec::with_capacity(self.cols * self.rows);

        for i in 0..self.cols * self.rows {
            new_data[i] = self.data[i] * m;
        }

        return Matrix {
            cols: *self.cols,
            rows: *self.rows,
            data: new_data,
        };
    }
}

fn main() {}

我仍在熟悉 Rust 和系统编程，我确信错误非常明显。编译器告诉我:

error[E0053]: method `mul` has an incompatible type for trait
  --> src/main.rs:22:5
   |
22 |     fn mul(&self, m: f32) -> Matrix {
   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected struct `Matrix`, found &Matrix
   |
   = note: expected type `fn(Matrix, f32) -> Matrix`
              found type `fn(&Matrix, f32) -> Matrix`

它指的是 for 循环的内容(我相信)。我试过玩其他一些东西，但我无法理解它。

Answer 1

错误信息在这里:

error[E0053]: method `mul` has an incompatible type for trait
  --> src/main.rs:22:5
   |
22 |     fn mul(&self, m: f32) -> Matrix {
   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected struct `Matrix`, found &Matrix
   |
   = note: expected type `fn(Matrix, f32) -> Matrix`
              found type `fn(&Matrix, f32) -> Matrix`

让我们看看Mul trait 看看为什么你的实现不匹配:

pub trait Mul<RHS = Self> {
    type Output;
    fn mul(self, rhs: RHS) -> Self::Output;
}

这表示除非您进一步指定任何内容，否则 RHS 将与 Self 的类型相同。 Self 是特征将被实现的类型。让我们看看您的定义:

impl Mul<f32> for Matrix {
    type Output = Matrix;

    fn mul(&self, m: f32) -> Matrix {}
}

在您的情况下，您已将 f32 替换为 RHS，将 Matrix 替换为 Output。此外，Matrix 是实现类型。让我们采用特征定义并代入，生成一些伪 Rust:

pub trait Mul {
    fn mul(self, rhs: f32) -> Matrix;
}

现在你看到什么不同了吗？

// Trait
fn mul(self,  m: f32) -> Matrix;
// Your implementation
fn mul(&self, m: f32) -> Matrix;

您错误地指定了您使用 &self 而不是 self。

为完整起见，这里是实现。我免费修复了样式!

impl Mul<f32> for Matrix {
    type Output = Matrix;

    fn mul(self, m: f32) -> Matrix {
        let new_data = self.data.into_iter().map(|v| v * m).collect();

        Matrix {
            cols: self.cols,
            rows: self.rows,
            data: new_data,
        }
    }
}

这有点低效，因为它会释放和重新分配 data 向量。由于您是按值获取 Matrix，我们可以就地编辑它:

impl Mul<f32> for Matrix {
    type Output = Matrix;

    fn mul(mut self, m: f32) -> Matrix {
        for v in &mut self.data {
            *v *= m
        }

        self
    }
}