in sparse_autoencoder/explanations.py [0:0]
def apply_batched(fn, iterable, bs): for batch in batched(iterable, bs): ret = fn(batch) assert len(ret) == len(batch) yield from ret