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