- 如果有101个样本,batch size是100,那么这个函数会返回两个batch,第一个是100,第二个是1。
- 如果想要强制每个batch 的数目是一样的那么就在batch 的时候设置
drop_remainder=True
,.batch()
中也有这个参数。这样101个样本,只会返回一个数目是100的batch,剩下的一个样本会被丢弃。
image.png
[tf]关于make_one_shot_iterator()的一些事
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...