def _write_by_chunk(dset, arrs):
# Note: arrs should be a generator for performance reasons.
assert isinstance(dset, Dataset)
# Start the data.
offset = 0
for arr in arrs:
n = arr.shape[0]
arr = arr[...]
# Match the shape of the chunk array with the dset shape.
assert arr.shape == (n,) + dset.shape[1:]
dset[offset:offset + n, ...] = arr
offset += arr.shape[0]
# Check that the copy is complete.
assert offset == dset.shape[0]
评论列表
文章目录