若API未标明支持情况则代表该API暂无验证结果，待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.utils.data.DataLoader |
是 |
支持fp32 |
torch.utils.data.Dataset |
是 |
支持fp16，fp32，fp64，int8，uint8，int16，int32，int64 |
torch.utils.data.IterableDataset |
是 |
|
torch.utils.data.TensorDataset |
是 |
|
torch.utils.data.StackDataset |
||
torch.utils.data.ConcatDataset |
是 |
|
torch.utils.data.ChainDataset |
是 |
|
torch.utils.data.Subset |
是 |
支持int64 |
torch.utils.data._utils.collate.collate |
是 |
支持int64 |
torch.utils.data.default_collate |
是 |
支持fp32，int64，bool |
torch.utils.data.default_convert |
是 |
|
torch.utils.data.get_worker_info |
是 |
|
torch.utils.data.random_split |
是 |
|
torch.utils.data.Sampler |
是 |
支持int64 |
torch.utils.data.SequentialSampler |
是 |
支持fp32 |
torch.utils.data.RandomSampler |
是 |
支持fp32 |
torch.utils.data.SubsetRandomSampler |
是 |
支持fp32 |
torch.utils.data.WeightedRandomSampler |
是 |
支持fp32 |
torch.utils.data.BatchSampler |
是 |
支持fp32 |
torch.utils.data.distributed.DistributedSampler |
是 |
支持int32 |