Skip to content

Commit 0613135

Browse files
kmikefmassa
authored andcommitted
improve make_grid docstring (pytorch#227)
* DOC fixed typo in make_grid argument name * DOC document "padding" make_grid argument
1 parent 77eb7ea commit 0613135

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

torchvision/utils.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,16 +10,17 @@ def make_grid(tensor, nrow=8, padding=2,
1010
Args:
1111
tensor (Tensor or list): 4D mini-batch Tensor of shape (B x C x H x W)
1212
or a list of images all of the same size.
13-
nrows (int, optional): Number of rows in grid. Final grid size is
13+
nrow (int, optional): Number of rows in grid. Final grid size is
1414
(B / nrow, nrow). Default is 8.
15+
padding (int, optional): amount of padding. Default is 2.
1516
normalize (bool, optional): If True, shift the image to the range (0, 1),
1617
by subtracting the minimum and dividing by the maximum pixel value.
1718
range (tuple, optional): tuple (min, max) where min and max are numbers,
1819
then these numbers are used to normalize the image. By default, min and max
1920
are computed from the tensor.
20-
scale_each(bool, optional): If True, scale each image in the batch of
21+
scale_each (bool, optional): If True, scale each image in the batch of
2122
images separately rather than the (min, max) over all images.
22-
pad_value(float, optional): Value for the padded pixels.
23+
pad_value (float, optional): Value for the padded pixels.
2324
2425
Example:
2526
See this notebook `here <https://gist.github.com/anonymous/bf16430f7750c023141c562f3e9f2a91>`_

0 commit comments

Comments
 (0)