PDEBench

该数据集包含基准数据，通过基于不同 PDE 的数值模拟生成，即 1D 平流、1D Burgers'、1D 和 2D 扩散反应、1D 扩散吸附、1D、2D 和 3D 可压缩 Navier-Stokes、2D 达西流、 和二维浅水方程。 该数据集旨在推进科学 ML 研究领域。 通常，数据以 HDF5 格式存储，数组维度按照约定 [b,t,x1,...,xd,v] 打包，其中 b 是批量大小（即样本数），t 是时间维度，x1,...,xd 是空间维度，v 是通道数（即感兴趣的变量数）。