mlx.core.import_function

mlx.core.import_function#

import_function(file: str) → Callable#

从文件中导入一个函数。

导入的函数可以使用 *args 和 **kwargs 调用，也可以使用一个包含数组的元组和/或包含字符串键和数组值的字典调用。导入的函数总是返回一个数组元组。

警告

这是实验性 API 的一部分，在未来版本的 MLX 中可能会发生变化。使用旧版本 MLX 导出的函数可能与未来版本不兼容。

参数:

file (str) – 要导入函数的文件的路径。

返回值:

导入的函数。

返回类型:

Callable

示例

>>> fn = mx.import_function("function.mlxfn")
>>> out = fn(a, b, x=x, y=y)[0]
>>>
>>> out = fn((a, b), {"x": x, "y": y}[0]
目录