diff --git a/torch/__init__.py b/torch/__init__.py index 3f2a2c0d9631..2c37ecac9a52 100644 --- a/torch/__init__.py +++ b/torch/__init__.py @@ -34,6 +34,8 @@ from typing import ( ) from typing_extensions import ParamSpec as _ParamSpec, TypeIs as _TypeIs +from . import version + if TYPE_CHECKING: from .types import Device, IntLikeType