torch.cuda.set_stream

torch.cuda.set_stream(stream) [source]

Устанавливает текущий поток. Это API-обёртка для установки потока.

Использование этой функции не рекомендуется в пользу контекстного менеджера stream.

Параметры

stream (Поток) – выбранный поток. Эта функция является пустой операцией, если этот аргумент None.