dataset.py 文件源码

def parse_record_fn(self):
    def _parse(proto):
      spec = self.record_spec()
      ctx, seq = tf.parse_single_sequence_example(proto, **spec)
      ctx.update(seq)
      result = []
      for field in self.FIELDS:
        result.append(ctx[field])
      return result
    return _parse