pytorch集锦(7)-处理数据DataLoader和Dataset(4)
默认情况下,如果钉扎逻辑看到一个自定义类型的批(如果您有一个返回自定义批类型的collate_fn,则会发生这种情况),或者如果批中的每个元素都是自定义类型,钉扎逻辑将无法识别它们,它将返回该批(或这些元素)而不钉扎内存。(请参见常见问题解答中的本节)。例如,如果每个数据样本由一个3通道图像和一个完整的类标签组成,即数据集的每个元素都返回一个元组(image,class_index),则默认的collated_fn将这些元组的列表整理成一个由批处理图像张量和批处理类标签tensor组成的单个元组