mlx.core.export_function

mlx.core.export_function#

导出函数(文件： str, 函数： 可调用的, *参数, 无形的： 布尔值 = 假, **关键字参数) → 无#

将函数导出到文件。

必须提供示例输入数组以导出函数。示例输入可以是变量*args和**kwargs，或者是数组的元组和/或带有数组值的字符串键的字典。

警告

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

Parameters：

  • 文件 (str) – 导出函数的文件路径。

  • fun (可调用) – 一个函数，它接受零个或多个 array 作为输入，并返回一个或多个 array。

  • *args (array) – 函数的示例数组输入。

  • shapeless (bool, 可选) – 函数是否允许输入具有可变形状。默认值：False。

  • **kwargs (array) – 函数的额外示例关键字数组输入。

示例

def fun(x, y):
    return x + y

x = mx.array(1)
y = mx.array([1, 2, 3])
mx.export_function("fun.mlxfn", fun, x, y=y)