concatenate feature maps pytorch

I am new in machine learning and I was trying to merge feature maps that I got from two different encoders which have different dimensions.

simclr_features: torch.Size([543, 512])
imagenet_features: torch.Size([543, 1024])

so I wanted: torch.Size([543, 1536]). what is a possible solution to do that?

>Solution :

You can use torch.cat:

>>> torch.cat((simclr_features, imagenet_features), dim=1)