torch.cuda.stream(stream)
作用
选择给定流的上下文管理器。
在其上下文中排队的所有CUDA核心将在所选流上入队。
参数
stream (Stream) – 所选流。如果是None,则这个管理器是无效的。
因为是个萌新小白, 第一次见到这个东西, 还不是很了解, 如果以后理解深刻了再来补充, 也希望能有前辈大神能够指出
torch.cuda.stream(stream)
选择给定流的上下文管理器。
在其上下文中排队的所有CUDA核心将在所选流上入队。
stream (Stream) – 所选流。如果是None,则这个管理器是无效的。
因为是个萌新小白, 第一次见到这个东西, 还不是很了解, 如果以后理解深刻了再来补充, 也希望能有前辈大神能够指出