title: How do I write the result of tf.Data.dataset.take() using TFRecordWriter?
body: <p>I am attempting to read in a tfrecord file, shuffle and split it, and then save to two tfrecords so I can use the same validation data in multiple runs.</p>
  
  <p>The problem I'm getting is that what I assumed would be a Dataset object is a TakeDataset object.</p>
  
  <p>I'm on Ubuntu 16.04 and using Tensorflow 2.1.0.</p>
  
  <p>The relevant code:</p>
  
  <pre><code>dataset = create_dataset(filenames_train, batch_size=25)
  
  validation_dataset = dataset.take(8400)
  training_dataset = dataset.skip(8400)
  
  writer_valid = tf.data.experimental.TFRecordWriter("./Data/valid.tfrecord")
  writer_valid.write(validation_dataset)
  writer_train = tf.data.experimental.TFRecordWriter("./Data/train.tfrecord")
  writer_train.write(training_dataset)
  </code></pre>
  
  <p>The output is:</p>
  
  <pre><code>...in &lt;module&gt;
      writer_valid.write(validation_dataset)
    File "/home/kyle/Desktop/learning/venv/lib/python3.5/site-packages/tensorflow_core/python/data/experimental/ops/writers.py", line 107, in write
      if not dataset_ops.get_structure(dataset).is_compatible_with(
  AttributeError: 'tuple' object has no attribute 'is_compatible_with'
  
  </code></pre>