使用流

指定 Stream

所有操作（包括随机数生成）都接受一个可选的关键字参数 stream。stream 参数指定了操作应该在哪个 Stream 上运行。如果未指定流，则操作将在默认设备的默认流上运行：mx.default_stream(mx.default_device())。stream 参数也可以是 Device（例如 stream=my_device），在这种情况下，操作将在提供的设备的默认流 mx.default_stream(my_device) 上运行。