torch.accelerator.current_stream

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

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

Параметры:

device (torch.device, str, int, необязательный) – указанное устройство, тип которого должен совпадать с типом текущего устройства ускорителя. Если не указано, по умолчанию используется torch.accelerator.current_device_index().

Возвращает:

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

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

torch.Stream