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

Expand composites in tf.test.TestCase.evaluate.

Browse Source 
This is helpful to allow CompositeTensors to be used with tf.test.TestCase.evaluate.

PiperOrigin-RevId: 313446907
Change-Id: I74ab70647b674689f36ea5738b57d44d02ec9685
This commit is contained in:
RJ Skerry-Ryan 2020-05-27 12:58:04 -07:00 committed by TensorFlower Gardener
parent 426869b50f
commit 49a5378fb1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/framework/test_util.py
View File

@ -2128,7 +2128,9 @@ class TensorFlowTestCase(googletest.TestCase):
values=tensor.values.numpy(),
indices=tensor.indices.numpy(),
dense_shape=tensor.dense_shape.numpy())
return tensor.numpy()
# Convert tensors and composite tensors to numpy arrays.
return nest.map_structure(lambda t: t.numpy(), tensor,
expand_composites=True)
except AttributeError as e:
six.raise_from(ValueError("Unsupported type %s." % type(tensor)), e)