megengine.data.dataset.ArrayDataset

class ArrayDataset(*arrays)[源代码]

适用于 Numpy ~.ndarray 数据的 Dataset 类。

需要一个或多个 NumPy 数组来初始化数据集,且表示样本数的维数应当一致。

方法