Skip to content

GitLab

Commit 8f6b9df8 authored by Dmitry Ulyanov's avatar Dmitry Ulyanov Committed by Soumith Chintala
Browse files

Make clearer about nrow parameter in make_grid (#123)

parent 510f095d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/utils.py
View file @ 8f6b9df8
...@@ -8,7 +8,7 @@ def make_grid(tensor, nrow=8, padding=2, ...@@ -8,7 +8,7 @@ def make_grid(tensor, nrow=8, padding=2,
""" """
Given a 4D mini-batch Tensor of shape (B x C x H x W), Given a 4D mini-batch Tensor of shape (B x C x H x W),
or a list of images all of the same size, or a list of images all of the same size,
makes a grid of images makes a grid of images of size (B / nrow, nrow).
normalize=True will shift the image to the range (0, 1), normalize=True will shift the image to the range (0, 1),
by subtracting the minimum and dividing by the maximum pixel value. by subtracting the minimum and dividing by the maximum pixel value.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment