torch.cuda.set\_stream ====================== .. currentmodule:: torch.cuda .. autofunction:: set_stream