Rust填充张量宏无法匹配元组变量的问题及解决问询

问题描述 #

实现了用于生成嵌套张量的宏fill_tensor：

macro_rules! fill_tensor {
    ($elem:expr; ($dim:expr, $($dims:expr),+)) => {
        vec![fill_tensor!($elem; ($($dims),+)); $dim]
    };
    ($elem:expr; $dim:expr) => {
        vec![$elem; $dim]
    };
}

调用fill_tensor!(4; (3_usize, 3_usize))可正常生成3×3矩阵，但调用let shape: (usize, usize) = (3_usize, 3_usize); fill_tensor!(4; shape)时，会报错：mismatched types: expected type usize, found tuple (usize, usize)。

原因分析 #

Rust的声明宏是语法模式匹配，而非类型匹配：

• 元组字面量(3_usize, 3_usize)是语法层面的「括号包裹多个表达式」结构，能匹配宏的第一个分支($elem:expr; ($dim:expr, $($dims:expr),+))。
• 变量shape是单个表达式（尽管它的类型是元组），宏会将其匹配到第二个分支($elem:expr; $dim:expr)，此时代码会尝试将整个元组作为Vec的长度参数，导致类型不匹配错误。

修复方案 #

由于声明宏无法基于类型匹配元组变量，推荐使用Trait抽象的方式实现，利用Rust的类型系统处理不同维度的元组：

实现代码 #

// 定义Trait，抽象张量填充逻辑
trait FillTensor<T> {
    type Output;
    fn fill_tensor(&self, elem: T) -> Vec<Self::Output>;
}

// 一维张量实现
impl<T: Clone> FillTensor<T> for usize {
    type Output = T;
    fn fill_tensor(&self, elem: T) -> Vec<T> {
        vec![elem; *self]
    }
}

// 二维张量实现
impl<T: Clone> FillTensor<T> for (usize, usize) {
    type Output = Vec<T>;
    fn fill_tensor(&self, elem: T) -> Vec<Vec<T>> {
        vec![self.1.fill_tensor(elem); self.0]
    }
}

// 三维张量实现（可按需扩展更多维度）
impl<T: Clone> FillTensor<T> for (usize, usize, usize) {
    type Output = Vec<Vec<T>>;
    fn fill_tensor(&self, elem: T) -> Vec<Vec<Vec<T>>> {
        vec![(self.1, self.2).fill_tensor(elem); self.0]
    }
}

fn main() {
    // 使用元组字面量生成二维张量
    let mat1 = (3_usize, 3_usize).fill_tensor(4);
    println!("{:?}", mat1);

    // 使用元组变量生成二维张量
    let shape: (usize, usize) = (3_usize, 3_usize);
    let mat2 = shape.fill_tensor(4);
    println!("{:?}", mat2);
}

方案优势 #

• 基于类型匹配，无论是元组字面量还是元组变量，都能根据其类型匹配对应的Trait实现。
• 可通过扩展Trait实现，支持任意维度的张量生成。
• 代码可读性更强，符合Rust的类型系统设计思路。

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