本篇博客主要讲解tf.data.Dataset的batch,repeat,shuffle函数,最后着重讲它们之间的顺序使用问题。
查询可用:tf.data.Dataset官方文档
内容参考:链接🔗
batch函数讲解
以下先创建一个Dataset类型
import tensorflow as tf
t = tf.range(10.)[:, None]
t = tf.data.Dataset.from_tensor_slices(t)
#<TensorSliceDataset shapes: (1,), types: tf.float32>
for i in t:
print(i.numpy())
#[0.]
#[1.]
#[2.]