src/transformers/models/swin2sr/modeling_swin2sr.py/Swin2SRPatchUnEmbeddings/forward
class Swin2SRPatchUnEmbeddings:    def forward(self, embeddings, x_size):
        batch_size, height_width, num_channels = embeddings.shape
        embeddings = embeddings.transpose(1, 2).view(batch_size, self.embed_dim, x_size[0], x_size[1])  # B Ph*Pw C
        return embeddings