[tf]关于make_one_shot_iterator()的一些事

  • 如果有101个样本,batch size是100,那么这个函数会返回两个batch,第一个是100,第二个是1。
  • 如果想要强制每个batch 的数目是一样的那么就在batch 的时候设置drop_remainder=True,.batch()中也有这个参数。这样101个样本,只会返回一个数目是100的batch,剩下的一个样本会被丢弃。
    image.png

你可能感兴趣的:([tf]关于make_one_shot_iterator()的一些事)