请用 PyTorch 实现一个 ImageDataset 类，继承自 Dataset 用于加载给定的图像数据集，并实现一个 collate_fn 函数，用于将数据集中的样本打包成批次。
考察说明
考察对 PyTorch Dataset 和 DataLoader collate_fn 机制的理解及编码实现能力
回答思路
- 正确实现 __init__、__len__、__getitem__ 方法
- 明确定义样本的返回格式（图像张量、标签）
- 实现有效的 collate_fn 处理不同尺寸或类型
- 考虑数据加载的边界情况和性能
本题已收录答题指导
本题附完整参考答案与评分标准
登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。