Check on the shape of the data and the expected shape of the data by the model file

[physnerds]

Overview

In the DNNROI inference, the model expects the data to have certain shape. Currently we dont have any checks to see if the shape of the incoming data and the what the model expects are compatible.
Add a torch centric function that does this check before proceeding forward.