mlx.core.stream

stream(s: Stream | Device) → mlx.core.StreamContext

Create a context manager to set the default device and stream.

参数:

s -- The Stream or Device to set as the default.

返回:

A context manager that sets the default device and stream.

Example: