单项选择题

若要在PyTorch 中构建自定义数据集,我们应继承哪个类?()

A.torch.utils.data.Dataset
B.torch.utils.data.DataLoader
C.torch.utils.data.TensorDataset
D.torch.utils.data.CustomDataset

微信扫码免费搜题