How to transform a tensor of a shape (224,224) to a tensor of shape (?,14,14)

Question

I have two classes, each one of them returns a tensor. In order to pass the tensor to the next function, it should be 3d (one dimension for the batch size). the first class retu