Add a util function for inference (predictions) #32
Description
Right now we use dataloader to load small subset of data during the test (inference). We need a util function to stack the subsets together. If needed another function to smooth edge effects.