抄一个 batch 函数

需求:将一个数组/可迭代的东西分成 batch

实现:三行代码

从 StackOverflow 上抄的

def chunk(it, size:int):
    it = iter(it)
    return iter(lambda: tuple(islice(it, size)), ())

蛮好用的


已发布

分类

作者:

标签

评论

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注