pytorch

1
import torch
2

3
from torch._subclasses.fake_tensor import (
4
    DynamicOutputShapeException,
5
    FakeTensor,
6
    FakeTensorMode,
7
    UnsupportedFakeTensorException,
8
)
9

10
from torch._subclasses.fake_utils import CrossRefFakeMode
11

12
__all__ = [
13
    "FakeTensor",
14
    "FakeTensorMode",
15
    "UnsupportedFakeTensorException",
16
    "DynamicOutputShapeException",
17
    "CrossRefFakeMode",
18
]
19