firstn
paddle.fluid.io.
firstn
(reader, n)[源代码]
该接口创建一个数据读取器,它可以返回的最大样本数为n。
- 参数:
- reader (callable) – 输入的数据读取器。
- n (int) – 可以返回的最大样本数。
返回: 新的的数据读取器。
返回类型: callable
- import paddle.fluid as fluid
- def reader():
- for i in range(100):
- yield i
- firstn_reader = fluid.io.firstn(reader, 5)
- for e in firstn_reader():
- print(e)
- # 输出结果为:0 1 2 3 4