How to plot different channels in grid?

vision
1

Hi All,
If I have tensor [1,16,16,16] i.e., 1 batch, 16 channels and image size 16 on 16. How is it possible to show plot in which each channel can be see separately?
Img_0 - tensor[:,0,:,:]
Img_1 - tensor[:,1,:,:]

Thank you!

2

You could use matplotlib and plot each channel in a subplot:

imgs = torch.empty(1, 16, 16, 16).uniform_(0, 1)

f, axarr = plt.subplots(imgs.size(1))

for idx in range(imgs.size(1)):
    img = imgs[:, idx].squeeze().numpy()
    axarr[idx].imshow(img)