StreamContext
class torch.cuda.StreamContext(stream)
Context-manager that selects a given stream.
All CUDA kernels queued within its context will be enqueued on a
   selected stream.
Parameters:
      Stream (Stream) -- selected stream. This manager is a no-
      op if it's "None".
Note:
 Streams are per-device.
