tensorflow-tf.data

102 阅读1分钟

tf.data是tensorflow中用于读取和写入数据的工具

相关API

1. tf.data.dataset

1. dataset.shuffle()-数据洗牌

1. 参数和返回值
  • buffer_size 缓冲区的大小,缓冲区大小是内存中一次可以容纳的item个数,洗牌时,利用堆排序的思想,进行洗牌.一般来讲,在内存允许的情况下,肯定时缓冲区越大越好.
  • seed=none
  • return 经过洗牌之后的tf.data.dataset对象