torch.cuda.default_stream

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

Возвращает поток Stream по умолчанию для указанного устройства.

Параметры:

device (torch.device или int, необязательный) – выбранное устройство. Возвращает поток Stream по умолчанию для текущего устройства, определяемого с помощью current_device(), если device равно None (по умолчанию).

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

Stream