torch.cuda.current_stream
torch.cuda.current_stream(device=None)
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 by "current_device()", if "device" is "None"
      (default).
Return type:
      Stream