本文主要是介绍pytorch中default_collate问题,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
output:
返回的都是列表的类型,但是第二个经过default_collate处理后,列表中的元素的类型变成了tensor的形式。
default_collate的用在地方,在我们试用torch.utils.data.DataLoder()数据后,当数据中有损坏的文件,我们则需要剔除,剔除试用的是filter(function,iteration)来处理,返回一个迭代器对象,我们可以通过list转换为列表的形式。但是比如送到神经网络中,我们还是需要是batch的形式,二原来的DataLoader处理后的图像就是tensor的形式。具体的可以看该函数,函数中的关键字里有transform,所以我们需要将处理后的数据,在转换为batch的形式,其实就是将列表中中的元素处理为tensor的形式。
另外需要记住的是:
default_collate来自于哪个包里。
这篇关于pytorch中default_collate问题的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!