Your PyTorch model should raise an error, if you try to pass the image tensor without a batch dimension. I don’t know, how CoreML handles it and you might get a better answer in their repository or discussion board, if it exists.