torch.cuda.default_stream¶
- torch.cuda.default_stream(device=None)[source][source]¶
Return the default
Streamfor a given device.- Parameters
device (torch.device or int, optional) – selected device. Returns the default
Streamfor the current device, given bycurrent_device(), ifdeviceisNone(default).- Return type
