mlx.core.async_eval

async_eval(*args)

异步评估一个 array 或一个 array 树。

注意

这是一个实验性 API，未来版本中可能会发生变化。

参数:

*args (数组 或 数组树) – 每个参数可以是一个单独的数组或一个数组树。如果给定的是树，节点可以是 Python list、tuple 或 dict。非数组的叶子节点将被忽略。

示例

>>> x = mx.array(1.0)
>>> y = mx.exp(x)
>>> mx.async_eval(y)
>>> print(y)
>>>
>>> y = mx.exp(x)
>>> mx.async_eval(y)
>>> z = y + 3
>>> mx.async_eval(z)
>>> print(z)