torch.cuda.stream
torch.cuda.stream(stream)
Wrapper around the Context-manager StreamContext that selects a
   given stream.
Parameters:
      stream (Stream) -- selected stream. This manager is a no-
      op if it's "None".
Return type:
      StreamContext
..Note:: In eager mode stream is of type Stream class while in JIT
   it is an object of the custom class "torch.classes.cuda.Stream".