diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi index 502113f1ce9a..d4b264a99b84 100644 --- a/stubs/tensorflow/tensorflow/__init__.pyi +++ b/stubs/tensorflow/tensorflow/__init__.pyi @@ -99,6 +99,7 @@ class Tensor: @property def op(self) -> Operation: ... def numpy(self) -> AnyArray: ... + def __array__(self, dtype: DTypeLike | None = None) -> AnyArray: ... def __int__(self) -> int: ... def __abs__(self, name: str | None = None) -> Tensor: ... def __add__(self, other: TensorCompatible) -> Tensor: ...