Example of collate_fn function and batch creation?

Hi! Would it be possible to get an example of `collate_fn` function for both a classification and segmentation (both single and multiple time steps) examples please? If you could provide an end to end example to getting a batch that would be highly appreciated as well. Many thanks.