torch.cpu.current_stream

torch.cpu.current_stream(device=None) [source]

Возвращает текущий выбранный Stream для данного устройства.

Параметры

device (torch.device или int, необязательно) – Игнорируется.

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

Поток

Примечание. Эта функция существует только для упрощения кода, не зависящего от устройства.