set_stream

paddle.device. set_stream ( stream=None )

设置当前的 stream。

参数

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

返回

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

代码示例

COPY-FROM: paddle.device.set_stream