生成tfrecords格式数据和使用dataset API使用tfrecords数据