如何在AssemblyScript中编写支持两矩阵输入输出的矩阵乘法函数？

算力阿峰

2026-10-20

AssemblyScript 矩阵乘法实现方案

我们可以基于行优先存储的一维Float64Array实现矩阵乘法，核心是建立二维矩阵坐标到一维数组索引的映射关系，具体实现如下：

前置约定

矩阵采用行优先顺序存储到一维Float64Array中，即逐行拼接所有元素
矩阵实例统一封装行、列数和存储数据，避免维度不匹配问题
矩阵乘法合法性前提：左矩阵的列数 = 右矩阵的行数

实现代码

// 矩阵封装类
class Matrix {
  rows: i32
  cols: i32
  data: Float64Array

  constructor(rows: i32, cols: i32, data: Float64Array | null = null) {
    this.rows = rows
    this.cols = cols
    const expectLength = rows * cols
    if (data) {
      if (data.length !== expectLength) throw new Error("矩阵数据长度与维度不匹配")
      this.data = data
    } else {
      this.data = new Float64Array(expectLength)
    }
  }
}

// 矩阵乘法函数
function matrixMultiply(left: Matrix, right: Matrix): Matrix {
  // 维度合法性校验
  if (left.cols !== right.rows) throw new Error("维度不匹配：左矩阵列数需等于右矩阵行数")

  const resultRows = left.rows
  const resultCols = right.cols
  const result = new Matrix(resultRows, resultCols)

  // 三重循环计算乘积累加
  for (let i: i32 = 0; i < resultRows; i++) {
    for (let j: i32 = 0; j < resultCols; j++) {
      let sum: f64 = 0
      for (let k: i32 = 0; k < left.cols; k++) {
        // 二维坐标转一维索引：行号 * 总列数 + 列号
        sum += left.data[i * left.cols + k] * right.data[k * right.cols + j]
      }
      result.data[i * resultCols + j] = sum
    }
  }

  return result
}