firstn

该接口创建一个数据读取器,它可以返回的最大样本数为n。

  • 参数:
    • reader (callable) – 输入的数据读取器。
    • n (int) – 可以返回的最大样本数。

返回: 新的的数据读取器。

返回类型: callable

  1. import paddle.fluid as fluid
  2. def reader():
  3. for i in range(100):
  4. yield i
  5. firstn_reader = fluid.io.firstn(reader, 5)
  6. for e in firstn_reader():
  7. print(e)
  8. # 输出结果为:0 1 2 3 4