[tf.contrib.data] map expects a nested structure.

Fixes #11786

PiperOrigin-RevId: 163359134
This commit is contained in:
A. Unique TensorFlower 2017-07-27 10:37:50 -07:00 committed by TensorFlower Gardener
parent ddd8e21b7c
commit 2139e7d8b1

View File

@ -548,8 +548,8 @@ labels = [0, 37, 29, 1, ...]
dataset = tf.contrib.data.Dataset.from_tensor_slices((filenames, labels))
dataset = dataset.map(
lambda filename, label: tf.py_func(
_read_py_function, [filename, label], [tf.uint8, label.dtype]))
lambda filename, label: tuple(tf.py_func(
_read_py_function, [filename, label], [tf.uint8, label.dtype])))
dataset = dataset.map(_resize_function)
```