评价此页

torch.accelerator.set_stream#

torch.accelerator.set_stream(stream)[source]#

将当前流设置为给定的流。

参数

stream (torch.Stream) – 给定的流,该流必须与当前加速器设备类型匹配。

注意

此函数会将当前设备索引设置为给定流的设备索引。