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

Update docs for tf.data.DatasetSpec.

Browse Source 
PiperOrigin-RevId: 271242288
This commit is contained in:
Andrew Audibert 2019-09-25 17:23:29 -07:00 committed by TensorFlower Gardener
parent 6a20128c5b
commit 7edb83de75
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/python/data/ops/dataset_ops.py
View File

@ -2561,7 +2561,14 @@ def to_variant(dataset):
"data.DatasetSpec", "data.DatasetSpec",
v1=["data.DatasetSpec", "data.experimental.DatasetStructure"]) v1=["data.DatasetSpec", "data.experimental.DatasetStructure"])
class DatasetSpec(type_spec.BatchableTypeSpec): class DatasetSpec(type_spec.BatchableTypeSpec):
"""Type specification for `tf.data.Dataset`.""" """Type specification for `tf.data.Dataset`.
See `tf.TypeSpec` for more information about TensorFlow type specifications.
>>> dataset = tf.data.Dataset.range(3)
>>> tf.data.DatasetSpec.from_value(dataset)
DatasetSpec(TensorSpec(shape=(), dtype=tf.int64, name=None), TensorShape([]))
"""
__slots__ = ["_element_spec", "_dataset_shape"] __slots__ = ["_element_spec", "_dataset_shape"]
@ -2598,6 +2605,7 @@ class DatasetSpec(type_spec.BatchableTypeSpec):
@staticmethod @staticmethod
def from_value(value): def from_value(value):
"""Creates a `DatasetSpec` for the given `tf.data.Dataset` value."""
return DatasetSpec(value.element_spec) # pylint: disable=protected-access return DatasetSpec(value.element_spec) # pylint: disable=protected-access
def _batch(self, batch_size): def _batch(self, batch_size):