如何用Rust的nalgebra crate高效初始化指定尺寸的3D矩阵

依赖配置 #

首先在Cargo.toml中添加nalgebra依赖，开启array特性以支持高维数组：

nalgebra = { version = "0.32", features = ["array"] }

实现代码 #

直接使用Array3::from_element方法即可完成初始化，该方法会分配连续内存空间，比嵌套Vec的内存布局更高效：

use nalgebra::Array3;

fn main() {
    let nx: usize = 100;  // x-resolution
    let ny: usize = 400;  // y-resolution
    let nl: usize = 4000; // z-resolution

    // 初始化维度为 (nx, ny, nl) 的全1 3D数组
    let f = Array3::from_element((nx, ny, nl), 1.0);
}

为什么比嵌套Vec更好 #

• 内存连续：Array3的所有元素存储在一块连续内存中，远优于嵌套Vec的分散内存布局，缓存命中率更高，访问速度更快
• API丰富：nalgebra提供了大量内置的张量运算、切片、变形方法，后续处理3D数据时无需手动实现复杂逻辑
• 代码简洁：一行代码完成初始化，避免了嵌套vec!宏的繁琐写法

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