diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-16 18:18:35 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-16 18:22:39 -0800 |
commit | 128572c316e6f2eb6346f920314ef98e88e75069 (patch) | |
tree | 81adea27d7149e711df54396801647d488834521 | |
parent | 090bb9168cbcb5bbb3d7fb8e0b64f7d00013d188 (diff) |
Adds a `shape` property to LabeledTensor.
#labeledtensor
PiperOrigin-RevId: 186073035
-rw-r--r-- | tensorflow/contrib/labeled_tensor/python/ops/core.py | 4 | ||||
-rw-r--r-- | tensorflow/contrib/labeled_tensor/python/ops/core_test.py | 3 |
2 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/labeled_tensor/python/ops/core.py b/tensorflow/contrib/labeled_tensor/python/ops/core.py index abc18aa123..0c6bba758b 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/core.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/core.py @@ -362,6 +362,10 @@ class LabeledTensor(object): return self._tensor.dtype @property + def shape(self): + return self._tensor.shape + + @property def name(self): return self._tensor.name diff --git a/tensorflow/contrib/labeled_tensor/python/ops/core_test.py b/tensorflow/contrib/labeled_tensor/python/ops/core_test.py index e70b492374..e378db56af 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/core_test.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/core_test.py @@ -244,6 +244,9 @@ class LabeledTensorTest(test_util.Base): def test_dtype(self): self.assertEqual(self.lt.dtype, self.lt.tensor.dtype) + def test_shape(self): + self.assertEqual(self.lt.shape, self.lt.tensor.shape) + def test_get_shape(self): self.assertEqual(self.lt.get_shape(), self.lt.tensor.get_shape()) |