[组合替代实现]torch.utils.set_module¶

torch.utils.set_module ¶

          torch.utils.set_module(obj, mod)

         

Paddle 无此 API，需要组合实现。

          # PyTorch 写法
torch.utils.set_module(obj, mod)

# Paddle 写法
obj.__module__ = mod