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: