5、加速Nx代码与机器学习数学基础

加速Nx代码与机器学习数学基础

1. 加速Nx代码：后端与编译器

在使用Nx时，有两种加速代码的基本方法：后端（backends）和编译器（compilers）。后端和编译器的关系类似于解释型编程语言和编译型编程语言的关系。

• 后端（Backends） ：Nx后端是Nx库的实现，它会立即计算Nx函数。默认的Nx后端是 Nx.BinaryBackend ，它使用纯Elixir来操作张量。后端每次都会计算Nx函数并产生结果，不具备融合或其他编译器优化的可能性。虽然速度较慢，但可以更快速地进行原型开发，因为不需要将代码组织成模块和数值定义，使用起来也更加灵活，无需遵循严格的数值定义编程模型。
  • 设置默认后端的方法：
    • 使用 Nx.default_backend/1 函数：

Nx.default_backend(EXLA.Backend)

    - 在应用配置中设置：

config :nx, default_backend: EXLA.Backend

• 编译器（Compilers） ：编译器实现了多阶段编程模型，通常性能更好，但需要更严格的编