output:
返回的都是列表的类型,但是第二个经过default_collate处理后,列表中的元素的类型变成了tensor的形式。
default_collate的用在地方,在我们试用torch.utils.data.DataLoder()数据后,当数据中有损坏的文件,我们则需要剔除,剔除试用的是filter(function,iteration)来处理,返回一个迭代器对象,我们可以通过list转换为列表的形式。但是比如送到神经网络中,我们还是需要是batch的形式,二原来的DataLoader处理后的图像就是tensor的形式。具体的可以看该函数,函数中的关键字里有transform,所以我们需要将处理后的数据,在转换为batch的形式,其实就是将列表中中的元素处理为tensor的形式。
另外需要记住的是:
default_collate来自于哪个包里。