tf Dataset API
https://zhuanlan.zhihu.com/p/30751039
https://zhuanlan.zhihu.com/p/37106443
关于其中shuffle时的buffer_size,可参考:
https://juejin.im/post/5b855d016fb9a01a1a27d035
https://user-gold-cdn.xitu.io/2018/8/28/16580f396628e48b
另外就是shuffle要放在map之前,否则会很费时间