pytorch/torch/_subclasses/__init__.py
2022-05-31 16:27:06 +00:00

10 lines
182 B
Python

import torch
from torch._subclasses.fake_tensor import FakeTensor, _device_not_kwarg_ops
__all__ = [
"FakeTensor",
"_device_not_kwarg_ops",
"_is_tensor_constructor",
]