diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py index c139a9b11de..f43663d5396 100644 --- a/tensorflow/python/framework/ops.py +++ b/tensorflow/python/framework/ops.py @@ -1067,15 +1067,17 @@ class _EagerTensorBase(Tensor): except core._NotOkStatusException as e: six.raise_from(core._status_to_exception(e.code, e.message), None) + def __array__(self): + return self._numpy() + def _numpy_internal(self): raise NotImplementedError() def _numpy(self): - # pylint: disable=protected-access try: return self._numpy_internal() - except core._NotOkStatusException as e: - six.raise_from(core._status_to_exception(e.code, e.message), None) + except core._NotOkStatusException as e: # pylint: disable=protected-access + six.raise_from(core._status_to_exception(e.code, e.message), None) # pylint: disable=protected-access @property def dtype(self):