我有以下代码用于在pytorch中测试网络（取自pix2pix）：
for i, data in enumerate(dataset):
model.set_input(data)
model.test()
我尝试过更改它，以便在迭代多个元素时仅访问一个元素（任何）。以下：
data=dataset[0]
导致
TypeError：“ CustomDatasetDataLoader”对象不支持索引
其中CustomDatasetDataLoader是custom class by the author的torch.utils.data.DataLoader与this的联系。
我对python还是很陌生，如果可能的话，如何为dataset的单个特定元素编制索引？