title: Access one image of dataset obtained with tfds.load()
body: <p>I download my dataset like this: <code>dataset = tfds.load('cifar10', split='train', shuffle_files=True)</code>. After applying the <code>batch()</code> method I have that <strong>dataset</strong> is:</p>
  <p><code>Attributes of object: {'image': &lt;tf.Tensor 'IteratorGetNext:0' shape=(128, 32, 32, 3) dtype=uint8&gt;, 'label': &lt;tf.Tensor 'IteratorGetNext:1' shape=(128,) dtype=int64&gt;}</code></p>
  <p>If I use <code>next(iter(train_dataset))</code> or <code>for elem in train_dataset.take(1): print (elem.numpy())</code> I need <code>eager_execution</code>.</p>
  <p>If I try <code>dataset[image]</code>, I get that the object is not subscriptable.</p>
  <p>If I try with:</p>
  <pre class="lang-py prettyprint-override"><code>iterator=dataset.make_one_shot_iterator()
    next_val = iterator.get_next()
    print(f'my_image: {next_val}')
    print(f'my_image: {next_val['image']}')
  </code></pre>
  <p>I get:</p>
  <pre><code>    print(f'my_image: {next_val['image']}')
                                   ^
  SyntaxError: invalid syntax
  </code></pre>
  <p>Even if I enable eager execution and do <code>batch = iter(dataset).get_next()</code> I get</p>
  <pre><code>Attributes of object: {'image': &lt;tf.Tensor: id=156, shape=(128, 32, 32, 3), dtype=uint8, numpy=
  array([[[[ 53,  53,  60],
           [ 63,  62,  66],
           [ 74,  72,  75],
           ...,
           [187, 194, 232],
  </code></pre>
  <p>but I can't access the single image.</p>
  <p>EDIT: I'm using tensorflow 1.15.5</p>