DataLoader¶
Collate¶
|
Take in a batch of data and put the elements within the batch into a tensor or ta_tensor with an additional outer dimension - batch size if relevant. |
Back to top