torch.cuda.set_stream

torch.cuda.set_stream(stream)[source][source]

设置当前流。这是一个包装 API，用于设置流。

建议使用 stream 上下文管理器来使用此函数。

参数:

流 (Stream) – 已选流。如果此参数为 None ，则此函数为空操作。