python - 如何在 pytorch 中为 DataLoader 提供图像和标签？

Question

我有一个 2D 张量，其中包含我的帧信息和一个 1D 张量，其中包含与每个帧相关的输出或标签信息。我创建一个函数并返回（框架、标签）。然后我把这两个给“torch.utils.data.DataLoader()”。但 DataLoader 不适用于两者（框架、标签），只能迭代其中之一。

标签： pythonpytorch