experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable __array__ access on EagerTensors (forwarding to _numpy)

Browse Source 
PiperOrigin-RevId: 309881480
Change-Id: I095a3ab0e3a991074d5f3949501302792cc9fd3b
This commit is contained in:
Martin Wicke 2020-05-04 22:08:41 -07:00 committed by TensorFlower Gardener
parent c265e89afa
commit f5ad74a467
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/framework/ops.py
View File

@ -1067,15 +1067,17 @@ class _EagerTensorBase(Tensor):
except core._NotOkStatusException as e: except core._NotOkStatusException as e:
six.raise_from(core._status_to_exception(e.code, e.message), None) six.raise_from(core._status_to_exception(e.code, e.message), None)
def __array__(self):
return self._numpy()
def _numpy_internal(self): def _numpy_internal(self):
raise NotImplementedError() raise NotImplementedError()
def _numpy(self): def _numpy(self):
# pylint: disable=protected-access
try: try:
return self._numpy_internal() return self._numpy_internal()
except core._NotOkStatusException as e: except core._NotOkStatusException as e: # pylint: disable=protected-access
six.raise_from(core._status_to_exception(e.code, e.message), None) six.raise_from(core._status_to_exception(e.code, e.message), None) # pylint: disable=protected-access
@property @property
def dtype(self): def dtype(self):