mlx.core.import_function

import_function(file: str) → Callable

Import a function from a file.

The imported function can be called either with *args and **kwargs or with a tuple of arrays and/or dictionary of string keys with array values. Imported functions always return a tuple of arrays.

警告

This is part of an experimental API which is likely to change in future versions of MLX. Functions exported with older versions of MLX may not be compatible with future versions.

参数:

file (str) -- The file path to import the function from.

返回:

The imported function.

返回类型:

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]