StreamContext

class torch.cuda.StreamContext(stream) [source]

Контекстный менеджер, который выбирает заданный поток.

Все ядра CUDA, помещённые в его контекст, будут помещены в выбранный поток.

Параметры:

Поток (Поток) – выбранный поток. Этот менеджер не выполняет никаких действий, если это None.

Примечание

Потоки зависят от устройства.