If you have a list of image tensors you can use this handy util from torchvision: torchvision.utils.make_grid torchvision.utils — Torchvision 0.10.0 documentation (pytorch.org) 1.
It takes in a list of image tensors and makes a nice grid of images for you, one of the most convienent utils I’ve used for displaying multiple images.