Collate function on libtorch

Hi,

Is there some example of using collate func for dataloader with custom dataset in libtorch?

This collate test might be useful.

Thanks, i will look into it