python - pytorch Dataset.__getitem__ 是否必须返回字典？

Question

这个问题在这里已经有了答案:





Understanding __getitem__ method

(7 个回答)


6 个月前关闭。




编辑:这不是关于一般 __getitem__方法但是__getitem__的用法在 Pytorch 数据集子类中
我正在尝试实现 Pytorchs Dataset-class 的使用。
指南例如 here真的很好，但我很难弄清楚 Pytorch 对 __getitem__ 返回值的要求.在 Pytorch documentation我找不到任何关于它应该返回什么的信息；是否有任何大小为 2 的可迭代对象，例如 [sample,target], (sample,target) ?在某些指南中，他们返回一个 dict，但他们没有指定它是否必须是一个返回的 dict。

Answer 1

PyTorch 对 DataSet 的返回值没有要求 __getitem__方法。它可以是任何东西，但您通常会遇到张量、张量元组、字典(例如 {'features':..., 'label':...} )等。
在 2d 数据中通常返回单个张量，其最后一列是目标值，但同样您可能会看到特征和目标的元组/字典明确分开。
请注意，不要求您返回两个值——在许多无监督的上下文中(例如自动编码器)，只有一组特征，没有明确的目标。