torch.xpu.set_stream¶ torch.xpu.set_stream(stream)[源代码][源代码]¶ 设置当前流。这是一个用于设置流的包装器 API。不鼓励使用此函数，而推荐使用 stream 上下文管理器。 参数 stream (Stream) – 选择的流。如果此参数为 None，则此函数不执行任何操作。
