torch.cuda.set_stream

torch.cuda.set_stream(stream) [source]

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

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

Параметры:

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