I am concatenating real images with labels and the output shape is [128, 11,28, 28]. How can I visualize this image?
encoded_labels = F.one_hot(reallabels, num_classes)
encoded_labels = encoded_labels [:, :, None, None]
encoded_labels = encoded_labels .repeat(1, 1, mnist_shape[1], mnist_shape[2]) 

Here the encoded_labels size is torch.Size([128, 10, 28, 28])
Now I want to concatenate it with images
cat_real_labels = torch.cat((realImages.float(), encoded_labels .float()),1)
output: torch.Size([128, 11, 28, 28])
How can I visualize this image?