Read TFRecord File into TF Dataset

Read-TFRecord.py

# Read a TFRecord file into a TF Dataset
def map_fn(serialized_example):
  feature = {
      'label': tf.io.FixedLenFeature([1], tf.int64),
      'ambient': tf.io.FixedLenFeature([16000], tf.float32),
      'target': tf.io.FixedLenFeature([4000], tf.float32)
  }
  example = tf.io.parse_single_example(serialized_example, feature)
  ambient = tf.expand_dims(example['ambient'], 1)
  target = tf.expand_dims(example['target'], 1)
  label = tf.cast(example['label'], tf.int32)
  return (ambient, target), label

dataset = tf.data.TFRecordDataset('example.tfrecord')
dataset = dataset.map(map_fn)
# dataset = dataset.repeat()
# dataset = dataset.shuffle(1000)
# dataset = dataset.batch(64)