使用流

指定

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