Make clearer about nrow parameter in make_grid (#123)

8f6b9df8 · Dmitry Ulyanov · Soumith Chintala · 510f095d · 8f6b9df8
Commit 8f6b9df8 authored Mar 23, 2017 by Dmitry Ulyanov Committed by Soumith Chintala Mar 23, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/utils.py torchvision/utils.py +1 -1

No files found.
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -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),
    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),
    by subtracting the minimum and dividing by the maximum pixel value.