|
import torch
|
|
import sys
|
|
import types
|
|
|
|
|
|
class VFModule(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__)
|