import torch | |
import sys | |
import types | |
class VFModule(types.ModuleType): | |
vf: types.ModuleType | |
def __init__(self, name): | |
super(VFModule, self).__init__(name) | |
self.vf = torch._C._VariableFunctions | |
def __getattr__(self, attr): | |
return getattr(self.vf, attr) | |
sys.modules[__name__] = VFModule(__name__) |