when i use model  witch i have trainde to predict .i just could conver the picture to tensor with the shape of CWH.But my data missing a batch dimension.So how to add the batch dimension when prediction.?Thank you very much.