pytorch Dataset.\uuuuu getitem\uuuuuuu是否必须返回dict？

Question

编辑：这不是关于一般的__getitem__方法，而是关于__getitem__在Pytorch数据集子类中的用法

我试图实现Pytorchs数据集类的用法。 例如here的指南非常好，但是我很难理解Pytorch对__getitem__返回值的要求。在Pytorch documentation中，我找不到关于它应该返回什么的任何信息；是否有任何对象的大小为2，例如[sample,target], (sample,target)的对象？在一些指南中，它们返回一个dict，但是它们没有指定它是否具有作为返回的dict

Answer 1

PyTorch对数据集的__getitem__方法的返回值没有要求。它可以是任何东西，但您通常会遇到张量、张量元组、字典（例如{'features':..., 'label':...}）等

在2d数据中，通常返回一个张量，其最后一列是目标值，但同样，您可能会看到特征和目标的元组/DICT显式分开

注意，不要求您返回两个值-在许多无监督的环境中（例如自动编码器），只有一组功能，没有明确的目标