torch.utils.set_module

torch.utils.set_module(obj, mod)

Set the module attribute on a python object for a given object for nicer printing