mlx.core.stream

stream(s: Union[Stream, Device]) → mlx.core.StreamContext

创建一个上下文管理器，用于设置默认设备和流。

参数：

s – 要设置为默认的 Stream 或 Device。

返回值：

一个上下文管理器，用于设置默认设备和流。

示例