megengine.data.Collator¶
-
class
Collator
[源代码]¶ 基类:
object
用于将一个列表中的样本合并成 mini-batch 形式的 Tensor. 方便从数据集中分批次加载数据。原始实现来自:https://github.com/pytorch/pytorch/blob/master/torch/utils/data/_utils/collate.py
方法
__init__
()初始化方法。
apply
(inputs)- 输入参数 inputs
sequence_N(tuple(CHW, C, CK))。