Currently, iter_torch_batches() provides the capability of automatic batching, just like what torch.utils.data.dataloader does by default [link]. However, users generally have more complicated input ...