标签: python pytorch
我有一个2D张量,其中包含我的帧信息,还有一个1D张量,其输出或标签信息与每个帧有关。我做一个函数并返回(框架,标签)。然后,将这两个信息提供给“ torch.utils.data.DataLoader()”。但是DataLoader不能同时用于这两个(框架,标签),而只能为其中之一进行迭代。