set_stream¶

参数¶

• stream (paddle.device.Stream) - 希望设置的 stream。

返回¶

paddle.device.Stream，设备之前的 stream。

代码示例¶

>>> import paddle

>>> paddle.set_device('custom_cpu')
>>> s = paddle.device.Stream()
>>> paddle.device.set_stream(s)