What is the correct way of using trchvision.utils.make_grid()

vision
4

They are in CHW. and here is the print output for
print(img.shape,recons.shape)
torch.Size([1, 28, 28]) torch.Size([1, 28, 28])