pytorch.utils.data.DataLoader:将自定义的Dataset根据batch size大小、是否shuffle等封装成一个Batch Size大小的Tensor,用于后面的训练,dataloader本质是一个可迭代对象,使用iter()访问,不能使用next()访问。
PyTorch数据处理工具箱utils
最新推荐文章于 2023-03-17 18:07:19 发布
pytorch.utils.data.DataLoader:将自定义的Dataset根据batch size大小、是否shuffle等封装成一个Batch Size大小的Tensor,用于后面的训练,dataloader本质是一个可迭代对象,使用iter()访问,不能使用next()访问。