torch.xpu.stream

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

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

Параметры:

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

Тип возвращаемого значения:

StreamContext