张量计算库的现状#

张量计算库的位置#

  • 张量计算库(多维数组库)被广泛使用于数值计算、数据科学、机器学习等领域。

  • 不同的张量计算库针对不同计算硬件进行了优化。

alt text

常见张量计算库#

alt text

张量化计算相关工作 / 软件#

  • IFEM (Chen, 2008)

  • torch-fenics (Barkman)

  • PyTorch-FEA (Liang, et al. 2023)

  • Magnum.np (Florian, et al. 2023)

  • JAX-FEM (Xue, et al. 2023)

  • Tensorly(Kossaifi,et al 2019)

  • ……

⚠️ 可能存在的问题#

  • 软件聚焦于单一领域或算法

  • 模块化设计仍有进步空间(提升可维护性、可扩展性)

  • 算法的张量化不完全