update docs for nrows param in make_grid (#250)

c94ea5ff · Alykhan Tejani · Francisco Massa · dd53c974 · c94ea5ff
Commit c94ea5ff authored Sep 08, 2017 by Alykhan Tejani Committed by Francisco Massa Sep 08, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/utils.py torchvision/utils.py +2 -2

No files found.
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -10,8 +10,8 @@ def make_grid(tensor, nrow=8, padding=2,
    Args:
        tensor (Tensor or list): 4D mini-batch Tensor of shape (B x C x H x W)
            or a list of images all of the same size.
-        nrow (int, optional): Number of rows in grid. Final grid size is
-            (B / nrow, nrow). Default is 8.
+        nrow (int, optional): Number of images displayed in each row of the grid.
+            The Final grid size is (B / nrow, nrow). Default is 8.
        padding (int, optional): amount of padding. Default is 2.
        normalize (bool, optional): If True, shift the image to the range (0, 1),
            by subtracting the minimum and dividing by the maximum pixel value.