Pytorch 에서 image grid

import matplotlib.pytplot as plt
import torchvision

for data in dataloader:
	img, label = data
    
    plt.imshow(torchvision.utils.make_grid(img, normalize).permute(1,2,0))
    plt.show()
    break

# make_grid(4차원 tensor)
# imshow(image.permute(1,2,0)) 채널을 마지막으로 옮겨야 됨