.. _cn_api_paddle_device_set_stream:

set_stream
-------------------------------

.. py:function:: paddle.device.set_stream(stream=None)

设置当前的 stream。


参数
::::::::::::

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

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

代码示例
::::::::::::
COPY-FROM: paddle.device.set_stream
