dataset.py 文件源码

def record_spec(self):
    field_dtype = lambda fname: tf.float32 if data_fields.FIELD_LOOKUP[fname].dtype==float \
        else tf.int64
    context_feats = {featname: tf.FixedLenFeature([], field_dtype(featname))
      for featname in context_fields}
    seq_feats = {featname: tf.FixedLenSequenceFeature([], field_dtype(featname))
        for featname in sequence_fields}
    return dict(context_features=context_feats, sequence_features=seq_feats)