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