mlx.core.import_function

mlx.core.import_function#

import_function(file: str) Callable#

從檔案匯入函式。

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]