torch.cuda.stream(stream)
Posted 怎样才能回到过去
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了torch.cuda.stream(stream)相关的知识，希望对你有一定的参考价值。
torch.cuda.stream(stream)作用
选择给定流的上下文管理器。
在其上下文中排队的所有CUDA核心将在所选流上入队。
参数
stream (Stream) – 所选流。如果是None，则这个管理器是无效的。
因为是个萌新小白, 第一次见到这个东西, 还不是很了解, 如果以后理解深刻了再来补充, 也希望能有前辈大神能够指出
以上是关于torch.cuda.stream(stream)的主要内容，如果未能解决你的问题，请参考以下文章