Rate this Page
★ ★ ★ ★ ★

torch.mps.current_stream#

torch.mps.current_stream()[source]#

Return the currently selected Stream.

Return type:

Stream

On this page
Show Source
PyTorch Libraries