我尝试合并从两个不同编码器（在Pytorch中具有不同维度）获得的特征图simclr_特征：torch.Size([543, 512])图像网络特征：torch.Size([543, 1024])个所以我想：torch.Size([543, 1536]) .什么是可能解决方案？
torch.Size([543, 512])
torch.Size([543, 1024])
torch.Size([543, 1536])
j91ykkif1#
您可以使用torch.cat：
torch.cat
torch.cat((simclr_features, imagenet_features), dim=1)
1条答案
按热度按时间j91ykkif1#
您可以使用
torch.cat：