megengine.data.dataset.ArrayDataset¶

class ArrayDataset(*arrays)[源代码]¶

基类：megengine.data.dataset.meta_dataset.Dataset

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

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

方法

__init__(*arrays)

初始化方法。

megengine.data.dataset.Dataset megengine.data.dataset.StreamDataset