train_data=TensorDataset(a,b) print(train_data[0:4]) 輸出結(jié)果如下: (tensor([[1, 1, 1], [2, 2, 2], [3, 3, 3], [4, 4, 4]]), tensor([1, 2, 3, 4])) DataLoader DataLoader本質(zhì)上就是一個(gè)iterable(跟python的內(nèi)置類(lèi)型list等一樣),并利用多進(jìn)程來(lái)加速batch data的處理,使用yield...
www.dbjr.com.cn/python/297746u...htm 2025-5-20