我可以给你一个参考的Python版本的视频dataloader: import torch from torch.utils.data import DataLoader
class VideoDataloader(DataLoader): def init(self, data, batch_size, shuffle=True): self.data = data self.batch_size = batch_size self.shuffle = shuffle
def __iter__(s