torch.utils.set\_module ======================= .. currentmodule:: torch.utils .. autofunction:: set_module