pytorch
1"""
2This makes the functions in torch._C._VariableFunctions available as
3torch._VF.<funcname>
4without mypy being able to find them.
5
6A subset of those functions are mapped to ATen functions in
7torch/jit/_builtins.py
8
9See https://github.com/pytorch/pytorch/issues/21478 for the reason for
10introducing torch._VF
11
12"""
13import sys14import types15
16import torch17
18
19class VFModule(types.ModuleType):20vf: types.ModuleType21
22def __init__(self, name):23super().__init__(name)24self.vf = torch._C._VariableFunctions25
26def __getattr__(self, attr):27return getattr(self.vf, attr)28
29
30sys.modules[__name__] = VFModule(__name__)31