Collator

class Collator

用于将一个列表中的样本合并成 mini-batch 形式的 Tensor. 方便从数据集中分批次加载数据。原始实现来自：https://github.com/pytorch/pytorch/blob/master/torch/utils/data/_utils/collate.py

上一页

Lighting

下一页

megengine.functional