mlx.core.compile

compile(fun: Callable, inputs: object | None = None, outputs: object | None = None, shapeless: bool = False) → Callable

返回一个编译后的函数，该函数产生与 fun 相同的输出。

参数:

• fun (Callable) – 一个函数，接受可变数量的 array 或 array 的树形结构，并返回可变数量的 array 或 array 的树形结构。

• inputs (list 或 dict, 可选) – 这些输入将在函数编译期间与 fun 的输入一起被捕获。inputs 可以是 list 或 dict，其中包含任意嵌套的列表、字典或数组。非 array 的叶节点将被忽略。默认值: None

• outputs (list 或 dict, 可选) – 这些输出将在编译后的函数中被捕获和更新。outputs 可以是 list 或 dict，其中包含任意嵌套的列表、字典或数组。非 array 的叶节点将被忽略。默认值: None

• shapeless (bool, 可选) – 使用启用 shapeless 选项编译的函数在输入形状改变时不会重新编译。并非所有函数都可以启用 shapeless 进行编译。尝试以这种方式编译此类函数会抛出异常。请注意，即使将 shapeless 设置为 True，改变任何输入的维度数量或类型仍会导致重新编译。默认值: False

返回值:

一个编译后的函数，其输入参数与 fun 相同，并返回相同的输出。

返回类型:

Callable