torchvision.utils.make_grid creates a single image containing all passed image tensors as a grid, which might be more convenient in case you want to display a few images.
1 Like
torchvision.utils.make_grid creates a single image containing all passed image tensors as a grid, which might be more convenient in case you want to display a few images.