- 如果有101个样本,batch size是100,那么这个函数会返回两个batch,第一个是100,第二个是1。
- 如果想要强制每个batch 的数目是一样的那么就在batch 的时候设置
drop_remainder=True
,.batch()
中也有这个参数。
[tf]关于make_one_shot_iterator()的一些事
猜你喜欢
转载自blog.csdn.net/weixin_33701564/article/details/87337838
今日推荐
周排行