diff options
| author |  Jiri Simsa <jsimsa@google.com> | 2017-11-13 13:36:50 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-13 13:41:54 -0800 |
| commit | 9642c1c0164ac87acfff2d52f82626eb38667dd8 (patch) | |
| tree | 17d1a03c2c059891486ae457072bddc1d4452025 | |
| parent | bac56b37be7736c9da9a3257696a9c1241327d60 (diff) | |
Re-introduce support for key-value records into tf.data.read_batch_features.
PiperOrigin-RevId: 175580235
| -rw-r--r-- | tensorflow/contrib/data/python/ops/readers.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/data/python/ops/readers.py b/tensorflow/contrib/data/python/ops/readers.py index 08d6a7a605..632082b5f1 100644 --- a/tensorflow/contrib/data/python/ops/readers.py +++ b/tensorflow/contrib/data/python/ops/readers.py @@ -171,6 +171,8 @@ def read_batch_features(file_pattern, dataset = reader(filenames, *reader_args) else: dataset = reader(filenames) + if dataset.output_types == (dtypes.string, dtypes.string): + dataset = dataset.map(lambda _, v: v) if num_epochs != 1: dataset = dataset.repeat(num_epochs) if randomize_input: |