Shortcuts

torch.cuda.stream

torch.cuda.stream(stream)[source]

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources