写一个Python版本的视频dataloader

我可以给你一个参考的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__(self):
# 将数据混洗
if self.shuffle:
random.shuffle(self.data)

# 从数据中获取batch_size大小的数据
for i in range(0, len(self.data), self.batch_size):
batch_data = self.data[i:i+self.batch_size]
yield batch_data