StreamContext

class torch.cuda.StreamContext(stream) [исходный код]

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

Все ядра CUDA, поставленные в очередь в его контексте, будут добавлены в очередь выбранного потока.

Параметры:

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

Примечание

Потоки относятся к отдельным устройствам.