torch.cuda.default_stream#

torch.cuda.default_stream(device=None)[source]#

Return the defaultStream for a given device.

Parameters

device (torch.device orint,optional) – selected device. Returnsthe defaultStream for the current device, given bycurrent_device(), ifdevice isNone(default).

Return type

Stream