mindspore/docs/api/api_python/dataset/mindspore.dataset.PaddedDat...

23 lines
799 B
ReStructuredText
Raw Normal View History

2021-11-27 16:09:05 +08:00
mindspore.dataset.PaddedDataset
================================
2021-11-20 11:18:45 +08:00
2021-11-27 16:09:05 +08:00
.. py:class:: mindspore.dataset.PaddedDataset(padded_samples)
使用用户提供的填充数据创建数据集。可用于在分布式训练时给原始数据集添加样本,使数据集能平均分配给不同的分片。
2021-11-20 11:18:45 +08:00
**参数:**
2021-11-27 16:09:05 +08:00
**padded_samples** (list(dict)): 用户提供的样本数据。
2021-11-20 11:18:45 +08:00
**异常:**
2021-11-27 16:09:05 +08:00
- **TypeError**`padded_samples` 的类型不为list。
- **TypeError**`padded_samples` 的元素类型不为dict。
- **ValueError**`padded_samples` 为空列表。
**样例:**
>>> import numpy as np
>>> data = [{'image': np.zeros(1, np.uint8)}, {'image': np.zeros(2, np.uint8)}]
>>> dataset = ds.PaddedDataset(padded_samples=data)