tf.data.Dataset.from_tensor_slices.shuffle.batch,求解释
tensorflow吧
全部回复
仅看楼主
level 1
ZuoenDeng 楼主
buffer_size = 60000
batch_size = 256
#批量化和打乱数据
train_dataset = tf.data.Dataset.from_tensor_slices(train_images).shuffle(buffer_size).batch(batch_size)
上面就是相关的三条代码,关于批量化和打乱数据那行代码,单个方法from_tensor_slices、shuffle、batch我都知道是什么意思,但是这样合在一起我就不知道它是什么意思了,求大佬详细解释一下。
我的理解:from_tensor_slices就是将元数据按第一维度切分嘛,大概理解就是把数据分成一条一条的;shuffle就是打乱数据顺序;batch就是把256条数据合并为1批;所以最后train_dataset中的数据就是打乱顺序之后的60000/256=234.375批数据,并且每批数据有256条样本数据;
我理解的对吗?
2022年05月10日 12点05分 1
1