torch.cpu.current_stream

torch.cpu.current_stream(device=None) [исходный код]

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

Параметры:

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

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

Stream

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