I have lowered torch dimension of shape torch.zeros([16, 3, 32, 32]) into [32,32,3] numpy array by
    img = image.squeeze(0).detach().cpu().numpy()
    print(img.shape) #(16, 3, 32, 32)
    img = img[0, :,:,:].transpose(1,2,0)
    print(img.shape # (32, 32, 3)