美文网首页
[tf]关于make_one_shot_iterator()的一

[tf]关于make_one_shot_iterator()的一

作者: VanJordan | 来源:发表于2019-01-20 18:30 被阅读22次
  • 如果有101个样本,batch size是100,那么这个函数会返回两个batch,第一个是100,第二个是1。
  • 如果想要强制每个batch 的数目是一样的那么就在batch 的时候设置drop_remainder=True,.batch()中也有这个参数。
    image.png

相关文章

网友评论

      本文标题:[tf]关于make_one_shot_iterator()的一

      本文链接:https://www.haomeiwen.com/subject/qfffjqtx.html