| commit | 128572c316e6f2eb6346f920314ef98e88e75069 | [log] [tgz] |
|---|---|---|
| author | A. Unique TensorFlower <gardener@tensorflow.org> | Fri Feb 16 18:18:35 2018 -0800 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Fri Feb 16 18:22:39 2018 -0800 |
| tree | 81adea27d7149e711df54396801647d488834521 | |
| parent | 090bb9168cbcb5bbb3d7fb8e0b64f7d00013d188 [diff] |
Adds a `shape` property to LabeledTensor. #labeledtensor PiperOrigin-RevId: 186073035
diff --git a/tensorflow/contrib/labeled_tensor/python/ops/core.py b/tensorflow/contrib/labeled_tensor/python/ops/core.py index abc18aa..0c6bba7 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/core.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/core.py
@@ -362,6 +362,10 @@ 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 e70b492..e378db5 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 @@ 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())