StreamContext

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

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

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

Параметры

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

Примечание

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