torch.cuda.current_stream
-
torch.cuda.
current_stream
(device=None)[source] Returns the currently selected
Stream
for a given device.- Parameters
device (torch.device or int, optional) – selected device. Returns the currently selected
Stream
for the current device, given bycurrent_device()
, ifdevice
isNone
(default).