抄一个batch函数


发布于

|

分类

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

实现:三行代码

从StackOverflow上抄的

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

蛮好用的


评论

发表回复

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