模板定义维度的Matrix类不同维度矩阵无法相乘的解决方法

你的问题核心在于成员版本的operator*只能接受与当前类同模板参数的Matrix对象，而矩阵乘法要求左矩阵的列数等于右矩阵的行数，且结果矩阵维度与原矩阵都不同，因此需要将乘法运算符改为模板成员函数，以支持不同维度的Matrix类型。

具体修改方案 #

1. 重构矩阵乘法运算符 #

将类中原有的矩阵乘法成员函数：

Matrix operator*(const Matrix& other) const;
Matrix& operator*=(const Matrix& other);

替换为以下模板版本：

// 矩阵乘法：N×M 矩阵 × M×K 矩阵 = N×K 矩阵
template <size_t K>
Matrix<N, K, T> operator*(const Matrix<M, K, T>& other) const;

// 注意：矩阵间乘法不适合重载 *= 运算符
// 因为 Matrix 的维度是编译期模板参数固定的，无法通过 *= 修改自身维度，因此应该移除矩阵版本的 *=

这样修改后，当你用Matrix<7,3>乘Matrix<3,5>时，编译器会推导模板参数K=5，返回一个Matrix<7,5>类型的结果，完全符合矩阵乘法的数学规则。

2. 修正其他成员函数的返回值（可选但必要） #

你的代码中还有几个不符合数学定义的地方，一并修正：

• 转置函数：N×M矩阵转置后应为M×N矩阵，因此返回值需要调整：

  Matrix<M, N, T> Transposed();
  

• 迹函数：只有方阵（N=M）才能求迹，建议添加编译期断言避免错误调用：

  T Trace() {
      static_assert(N == M, "Trace is only defined for square matrices");
      // 实现迹的计算逻辑
  }
  

3. 修改后的完整类框架 #

template <size_t N, size_t M, typename T = int64_t>
class Matrix {
 public:
  Matrix();
  Matrix(std::vector<std::vector<T>> const input);
  Matrix(const T elem);

  // 同维度矩阵运算
  Matrix operator+(const Matrix& other) const;
  Matrix& operator+=(const Matrix& other);
  Matrix operator-(const Matrix& other) const;
  Matrix& operator-=(const Matrix& other);

  // 标量乘法
  Matrix operator*(const T& other) const;
  Matrix& operator*=(const T& other);

  // 跨维度矩阵乘法（模板版本）
  template <size_t K>
  Matrix<N, K, T> operator*(const Matrix<M, K, T>& other) const;

  // 修正转置函数返回值
  Matrix<M, N, T> Transposed();

  // 修正迹函数，添加编译期检查
  T Trace() {
      static_assert(N == M, "Trace is only defined for square matrices");
      // 实现迹的计算
  }

  T& operator()(const size_t i, const size_t j) const;

  bool operator==(const Matrix& other) const;

 private:
  // 成员变量定义
};

关键逻辑说明 #

• 模板成员函数允许我们在类内部定义接受不同模板参数的Matrix对象，只要满足矩阵乘法的维度约束（左列数=右行数），编译器就能正确推导返回值的维度。
• 移除矩阵版本的operator*=是因为模板参数是编译期确定的，对象的维度无法在运行时改变，强行重载*=会导致逻辑矛盾（无法将N×M矩阵变成N×K矩阵）。

内容的提问来源于stack exchange，提问作者Kruzenger