使用串流

指定 Stream

所有運算（包含隨機數產生）都可接受選用的關鍵字參數 stream。stream 參數指定運算應在何種 Stream 上執行。若未指定，運算會在預設裝置的預設串流上執行：mx.default_stream(mx.default_device())。stream 也可以是一個 Device`（例如 ``stream=my_device`），此時運算會在所提供裝置的預設串流 mx.default_stream(my_device) 上執行。