本文簡要介紹python語言中 torchdata.datapipes.iter.Batcher
的用法。
用法:
class torchdata.datapipes.iter.Batcher(datapipe: IterDataPipe, batch_size: int, drop_last: bool = False, wrapper_class=DataChunk)
datapipe-可迭代DataPipe正在批量處理
batch_size-每批的大小
drop_last-如果最後一批未滿,可選擇刪除最後一批
wrapper_class-在屈服之前應用於每個批次(類型
List
)的包裝器,默認為DataChunk
創建小批量數據(函數名稱:
batch
)。如果drop_last
設置為True
,則外部尺寸將添加為batch_size
;如果drop_last
設置為False
,則最後一批的外部尺寸將添加為length % batch_size
。示例
>>> from torchdata.datapipes.iter import IterableWrapper >>> dp = IterableWrapper(range(10)) >>> dp = dp.batch(batch_size=3, drop_last=True) >>> list(dp) [[0, 1, 2], [3, 4, 5], [6, 7, 8]]
參數:
相關用法
- Python PyTorch BatchSampler用法及代碼示例
- Python PyTorch BatchNorm1d用法及代碼示例
- Python PyTorch BatchNorm3d用法及代碼示例
- Python PyTorch BatchNorm2d用法及代碼示例
- Python PyTorch Bernoulli用法及代碼示例
- Python PyTorch Binomial用法及代碼示例
- Python PyTorch BucketBatcher用法及代碼示例
- Python PyTorch BCELoss用法及代碼示例
- Python PyTorch Bilinear用法及代碼示例
- Python PyTorch BCEWithLogitsLoss用法及代碼示例
- Python PyTorch Beta用法及代碼示例
- Python PyTorch frexp用法及代碼示例
- Python PyTorch jvp用法及代碼示例
- Python PyTorch cholesky用法及代碼示例
- Python PyTorch vdot用法及代碼示例
- Python PyTorch ELU用法及代碼示例
- Python PyTorch ScaledDotProduct.__init__用法及代碼示例
- Python PyTorch gumbel_softmax用法及代碼示例
- Python PyTorch get_tokenizer用法及代碼示例
- Python PyTorch saved_tensors_hooks用法及代碼示例
- Python PyTorch positive用法及代碼示例
- Python PyTorch renorm用法及代碼示例
- Python PyTorch AvgPool2d用法及代碼示例
- Python PyTorch MaxUnpool3d用法及代碼示例
- Python PyTorch Tensor.unflatten用法及代碼示例
注:本文由純淨天空篩選整理自pytorch.org大神的英文原創作品 torchdata.datapipes.iter.Batcher。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。